Update app.py

app.py

@@ -6,6 +6,7 @@ from transformers import AutoModelForCausalLM, AutoProcessor
 import torch
 import subprocess
 from io import BytesIO
+import os
 
 # Install flash-attn
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
@@ -52,12 +53,11 @@ 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
-
+def load_image_from_file(file_path):
+    if os.path.exists(file_path):
+        return Image.open(file_path)
+    else:
+        raise FileNotFoundError(f"Image file not found: {file_path}")
 # Custom CSS
 custom_css = """
 <style>
@@ -243,12 +243,12 @@ with gr.Blocks(css=custom_css) as iface:
     
     gr.Examples(
         examples=[
-            "eqn1.png",  # Replace with actual URLs of math problem images
-            "eqn2.png"
+            os.path.join(os.path.dirname(__file__), "eqn1.png"),
+            os.path.join(os.path.dirname(__file__), "eqn2.png")
         ],
         inputs=input_image,
         outputs=output_text,
-        fn=lambda url: solve_math_problem(load_image_from_url(url)),
+        fn=solve_math_problem,
         cache_examples=True,
     )