Update app.py

app.py

@@ -1,47 +1,46 @@
 import gradio as gr
-from transformers import AutoProcessor, SamModel
+from transformers import SamProcessor, SamModel
 from PIL import Image
 import torch
 import numpy as np
 import traceback
 
-# CPU-freundliches Modell
+# SAM Modell laden (CPU-kompatibel)
 model_id = "facebook/sam-vit-base"
-processor = AutoProcessor.from_pretrained(model_id)
+processor = SamProcessor.from_pretrained(model_id)
 model = SamModel.from_pretrained(model_id)
 
 def segment_image(image):
     try:
         device = torch.device("cpu")
         model.to(device)
+
+        # Bild vorbereiten
         inputs = processor(images=image, return_tensors="pt").to(device)
 
         with torch.no_grad():
             outputs = model(**inputs)
 
+        # Die alte API erwartet NICHT das Schlüsselwort 'outputs'
         masks = processor.post_process_masks(
-            outputs=outputs,
-            original_sizes=[image.size[::-1]],
-            reshaped_input_sizes=[image.size[::-1]]
+            outputs.pred_masks.cpu(),
+            inputs["original_sizes"].cpu(),
+            inputs["reshaped_input_sizes"].cpu()
         )
 
-        if len(masks) == 0 or "masks" not in masks[0]:
-            return "Keine Maske erkannt."
-
-        mask_array = masks[0]["masks"][0][0].cpu().numpy()
-        mask_image = Image.fromarray((mask_array * 255).astype(np.uint8))
+        mask = masks[0][0][0].numpy()
+        mask_image = Image.fromarray((mask * 255).astype(np.uint8))
         return mask_image
 
     except Exception as e:
-        # Volle Fehlerausgabe anzeigen
         return f"Fehler:\n{traceback.format_exc()}"
 
 demo = gr.Interface(
     fn=segment_image,
     inputs=gr.Image(type="pil", label="Upload your fish image"),
-    outputs="text",
-    title="FishBoost Segment Anything (Meta SAM CPU Debug)",
-    description="Debug-Version: Zeigt genaue Fehlermeldung bei Problemen mit SAM."
+    outputs=gr.Image(type="pil", label="Segmented Output"),
+    title="FishBoost Segment Anything (Meta SAM - CPU Safe)",
+    description="Stable version for Hugging Face CPU runtime. Uses Meta's SAM model."
 )
 
 demo.launch()