Update app.py

app.py

@@ -14,7 +14,7 @@ def encode_image(image_path):
 def run_code(code, groq_api_key):
     try:
         # Setup Groq API client with the provided API key
-        llava_client = Groq(api_key=groq_api_key)
+        llamavision_client = Groq(api_key=groq_api_key)
         llama_client = Groq(api_key=groq_api_key)
 
         fig, ax = plt.subplots()
@@ -39,9 +39,9 @@ def run_code(code, groq_api_key):
         # Encode the image for Groq API
         base64_image = encode_image("plot.png")
 
-        # Sending the plot image to LLava API to get the description
-        llava_completion = llava_client.chat.completions.create(
-            model="llava-v1.5-7b-4096-preview",
+        # Sending the plot image to Llama Vision API to get the description
+        llamavision_completion = client.chat.completions.create(
+            model="llama-3.2-11b-vision-preview",
             messages=[{
                 "role": "user",
                 "content": [
@@ -59,8 +59,8 @@ def run_code(code, groq_api_key):
             stop=None,
         )
 
-        # Extract the LLava description from the API response
-        llava_description = llava_completion.choices[0].message.content
+        # 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
         llama_completion = llama_client.chat.completions.create(
@@ -87,7 +87,7 @@ def run_code(code, groq_api_key):
         for chunk in llama_completion:
             llama_description += chunk.choices[0].delta.content or ""
 
-        return img, llava_description, llama_description
+        return img, llamavision_description, llama_description
 
     except Exception as e:
         return None, f"Error: {str(e)}", None
@@ -115,13 +115,13 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     submit_btn = gr.Button("Submit")
 
     with gr.Row():
-        output_llava_text = gr.Textbox(label="Description from LLaVA", interactive=False)
+        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)
 
     submit_btn.click(
         fn=run_code,
         inputs=[code_input, api_key_input],
-        outputs=[output_image, output_llava_text, output_llama_text]
+        outputs=[output_image, output_llamavision_text, output_llama_text]
     )
 
 # Launch the interface