Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 from huggingface_hub import login
 
-# ✅ Login sécurisé avec le secret
+# 🔐 Login avec ton token Hugging Face (stocké comme secret)
 hf_token = os.environ.get("HUGGINGFACE_API_KEY")
 login(hf_token)
 
@@ -12,6 +12,7 @@ from PIL import Image
 
 model_id = "ContactDoctor/Bio-Medical-MultiModal-Llama-3-8B-V1"
 
+# ✅ Charger le modèle avec custom code autorisé
 model = AutoModelForCausalLM.from_pretrained(
     model_id,
     trust_remote_code=True,
@@ -19,11 +20,16 @@ model = AutoModelForCausalLM.from_pretrained(
     device_map="auto"
 )
 
-processor = AutoProcessor.from_pretrained(model_id)
+# ✅ Charger le processor avec trust_remote_code aussi
+processor = AutoProcessor.from_pretrained(
+    model_id,
+    trust_remote_code=True
+)
 
+# 🧠 Fonction principale d'inférence
 def generate_answer(image, question):
     if not question or question.strip() == "":
-        return "❌ Please enter a medical question."
+        question = "Please describe this medical image."
 
     prompt = f"### User: {question}\n### Assistant:"
     
@@ -38,6 +44,7 @@ def generate_answer(image, question):
     except Exception as e:
         return f"⚠️ Internal Error: {str(e)}"
 
+# 🎛️ Interface Gradio
 demo = gr.Interface(
     fn=generate_answer,
     inputs=[
@@ -49,4 +56,5 @@ demo = gr.Interface(
     description="Multimodal Medical Assistant: upload an image and ask a medical question."
 )
 
+# 🚀 Lancer l'app
 demo.launch()