Upload 2 files

app.py

@@ -36,10 +36,8 @@ demo = gr.Interface(
     outputs=gr.Textbox(label="Answer"),
     title="UK Public Transport Assistant",
     description="Upload an image of UK public transport signage (like train timetables or metro maps), and ask a question related to it. Powered by LLaVA-1.5.",
-    examples=[
-        # Only use examples if the example file exists
-        ["assets/example.jpg", "Where is platform 3?"] if os.path.exists("assets/example.jpg") else None
-    ]
+    # Fix examples format - either provide valid examples or set to None
+    examples=None  # Remove examples to avoid format errors
 )
 
 if __name__ == "__main__":

llava_inference.py

@@ -30,10 +30,11 @@ class LLaVAHelper:
                 legacy=True
             )
             
-            # Load model with same cache directory and more explicit parameters
+            # Load model with same cache directory - fix parameter order
             self.model, self.image_processor, _ = load_pretrained_model(
-                model_name,
-                None,
+                model_path=model_name,
+                model_name=model_name,  # Add the missing model_name parameter
+                model_base=None,
                 cache_dir="./model_cache",
                 load_8bit=False,
                 load_4bit=False,