Update app.py

app.py

@@ -35,11 +35,10 @@ caption_model = AutoModelForCausalLM.from_pretrained("microsoft/git-base").to(de
 def home():
     return {"status": "Online", "instruction": "Add /docs to the URL to test the model"}
 
-# --- 2. وظيفة المعالجة (نفس المنطق الذي نجح معكِ في كولاب) ---
+# --- 2. وظيفة المعالجة ---
 
 @app.post("/analyze")
 async def analyze_image(file: UploadFile = File(...)):
-    # قراءة الصورة المرفوعة
     data = await file.read()
     original_image = Image.open(io.BytesIO(data)).convert("RGB")
 
@@ -51,9 +50,9 @@ async def analyze_image(file: UploadFile = File(...)):
         for i, box in enumerate(r.boxes):
             label = r.names[int(box.cls)]
             conf_score = float(box.conf[0])
-            coords = box.xyxy[0].tolist() # [xmin, ymin, xmax, ymax]
+            coords = box.xyxy[0].tolist() 
 
-            # 2. عملية القص (Cropping) للجزء المكتشف
+            # 2. عملية القص (Cropping) 
             cropped_img = original_image.crop((coords[0], coords[1], coords[2], coords[3]))
 
             # 3. وصف الجزء المقصوص عبر موديل GIT
@@ -68,13 +67,12 @@ async def analyze_image(file: UploadFile = File(...)):
                 "description": detailed_desc
             })
 
-    # إذا لم يجد YOLO شيئاً، نصف الصورة كاملة
     if not integrated_results:
         inputs = processor(images=original_image, return_tensors="pt").to(device)
         generated_ids = caption_model.generate(pixel_values=inputs.pixel_values, max_length=40)
         general_desc = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
         return {
-            "message": "No specific objects detected by YOLO. Here is a general description.",
+            "message": "No specific objects detected. General description provided.",
             "general_description": general_desc
         }
 
@@ -83,8 +81,6 @@ async def analyze_image(file: UploadFile = File(...)):
         "results": integrated_results
     }
 
-# --- 3. تشغيل السيرفر ---
+# --- 3. تشغيل السيرفر (تم تصحيح الشرطات السفلية هنا) ---
 if name == "__main__":
-    # المنفذ 7860 هو المطلوب في Hugging Face
     uvicorn.run(app, host="0.0.0.0", port=7860)
-