Update app.py

app.py

@@ -5,6 +5,7 @@ import requests
 from transformers import AutoModelForCausalLM, AutoProcessor
 import torch
 import subprocess
+from io import BytesIO
 
 # Install flash-attn
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
@@ -51,6 +52,12 @@ def solve_math_problem(image):
     model.to('cpu')
     return response
 
+# Function to load image from URL
+def load_image_from_url(url):
+    response = requests.get(url)
+    img = Image.open(BytesIO(response.content))
+    return img
+
 # Custom CSS
 custom_css = """
 <style>
@@ -221,6 +228,7 @@ custom_html = """
 </div>
 """
 
+
 # Create the Gradio interface
 with gr.Blocks(css=custom_css) as iface:
     gr.HTML(custom_html)
@@ -235,12 +243,12 @@ with gr.Blocks(css=custom_css) as iface:
     
     gr.Examples(
         examples=[
-            ["example_math_problem1.jpg"],
-            ["example_math_problem2.jpg"]
+            "https://i.imgur.com/2Gwd3bN.jpg",  # Replace with actual URLs of math problem images
+            "https://i.imgur.com/wPw5YtB.jpg"
         ],
         inputs=input_image,
         outputs=output_text,
-        fn=solve_math_problem,
+        fn=lambda url: solve_math_problem(load_image_from_url(url)),
         cache_examples=True,
     )