Update app.py

app.py

@@ -111,9 +111,7 @@ def generate_and_detect(prompt):
     )
     if response.status_code != 200:
         return None, f"Generation failed: {response.text}", None
-    image_pil = Image.open(requests.get(response.url, stream=True).raw).convert('RGB') \
-        if hasattr(response, 'url') else Image.frombytes('RGB', (512, 512), response.content)
-    image_pil = Image.open(__import__('io').BytesIO(response.content)).convert('RGB')
+    image_pil = Image.open(io.BytesIO(response.content)).convert('RGB')
     result, gradcam_img = detect_image(np.array(image_pil))
     return image_pil, result, gradcam_img