Update app.py

app.py

@@ -62,9 +62,9 @@ def run_code(code, groq_api_key):
         # Extract the Llama Vision description from the API response
         llamavision_description = llamavision_completion.choices[0].message.content
 
-        # Sending the plot image to Llama 3.1 API to get the description
+        # Sending the plot image to Llama 3.2 API to get the description
         llama_completion = llama_client.chat.completions.create(
-            model="llama-3.1-70b-versatile",
+            model="llama-3.2-90b-text-preview"",
             messages=[
                 {
                     "role": "system",
@@ -82,7 +82,7 @@ def run_code(code, groq_api_key):
             stop=None,
         )
 
-        # Extract the Llama 3.1 description from the API response
+        # Extract the Llama 3.2 description from the API response
         llama_description = ""
         for chunk in llama_completion:
             llama_description += chunk.choices[0].delta.content or ""
@@ -116,7 +116,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     with gr.Row():
         output_llamavision_text = gr.Textbox(label="Description from Llama 3.2 Vision", interactive=False)
-        output_llama_text = gr.Textbox(label="Description from Llama 3.1", interactive=False)
+        output_llama_text = gr.Textbox(label="Description from Llama 3.2 Text", interactive=False)
 
     submit_btn.click(
         fn=run_code,