Update app.py

app.py

@@ -17,20 +17,22 @@ bnb_config = BitsAndBytesConfig(
     bnb_4bit_compute_dtype=torch.float16
 )
 
-# Load model
+# Load model with revision pinning
 model = AutoModel.from_pretrained(
     "ContactDoctor/Bio-Medical-MultiModal-Llama-3-8B-V1",
     quantization_config=bnb_config,
     device_map="auto",
     torch_dtype=torch.float16,
     trust_remote_code=True,
-    token=api_token
+    token=api_token,
+    revision="main"  # Pin to specific revision
 )
 
 tokenizer = AutoTokenizer.from_pretrained(
     "ContactDoctor/Bio-Medical-MultiModal-Llama-3-8B-V1",
     trust_remote_code=True,
-    token=api_token
+    token=api_token,
+    revision="main"  # Pin to specific revision
 )
 
 def analyze_input(image_data, question):
@@ -73,19 +75,18 @@ def analyze_input(image_data, question):
 demo = gr.Interface(
     fn=analyze_input,
     inputs=[
-        gr.Image(type="numpy", label="Medical Image"),  # Removed optional parameter
+        gr.Image(type="numpy", label="Medical Image"),
         gr.Textbox(label="Question", placeholder="Enter your medical query...")
     ],
     outputs=gr.JSON(label="Analysis"),
     title="Bio-Medical MultiModal Analysis",
     description="Ask questions with or without an image",
-    allow_flagging="never",
+    flagging_mode="never"  # Updated from allow_flagging
 )
 
-# Launch with API access enabled
+# Launch with simplified parameters
 demo.launch(
     share=True,
     server_name="0.0.0.0",
-    server_port=7860,
-    enable_queue=True
+    server_port=7860
 )