Update app.py

app.py

@@ -22,19 +22,28 @@ if not api_key:
 client = OpenAI(api_key=api_key)
 
 yolo_model = YOLO("yolov8s.pt")
-stable_diffusion = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
+stable_diffusion = StableDiffusionPipeline.from_pretrained(
+    "runwayml/stable-diffusion-v1-5",
+    torch_dtype=torch.float16,
+    safety_checker=None
+).to(device)
 stable_diffusion.to(device)
 nltk.download("punkt")
 
-summarizer = pipeline("summarization", model="facebook/bart-large-cnn", device=0 if torch.cuda.is_available() else -1)
+summarizer = pipeline(
+    "summarization", 
+    model="facebook/bart-large-cnn", 
+    device_map="auto"
+)
 
 @spaces.GPU
-def detect_objects(image_path):
-    results = yolo_model(image_path)
+def detect_objects(image):
+    image_array = np.array(image)
+    results = yolo_model(image_array)
     detected_objects = []
     for r in results:
         for box in r.boxes:
-            class_id = int(box.cls.item())  # Διορθώθηκε
+            class_id = int(box.cls.item())
             label = yolo_model.names[class_id]
             detected_objects.append(label)
     return detected_objects