Update app.py

app.py

@@ -4,11 +4,20 @@ import numpy as np
 from ultralytics import YOLO
 import easyocr
 import logging
+import torch.serialization  # Für Allowlist
 
 # Logging einrichten
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Allowlist für YOLOv8-Modell
+try:
+    logger.info("Adding ultralytics.nn.tasks.DetectionModel to PyTorch safe globals")
+    torch.serialization.add_safe_globals(['ultralytics.nn.tasks.DetectionModel'])
+except Exception as e:
+    logger.error("Failed to add safe globals: %s", str(e))
+    raise
+
 # Lade YOLOv8-Modell
 try:
     logger.info("Loading YOLOv8 model")
@@ -102,18 +111,4 @@ def analyze_image(image, prompt):
             return {"prompt": prompt, "description": "Error analyzing candles. Try a clearer screenshot with visible candles and prices."}
 
     else:
-        return {"prompt": prompt, "description": "Unsupported prompt. Use 'Was siehst du auf dem Bild?' or 'List last 8 candles with their colors'."}
-
-# Erstelle Gradio-Schnittstelle
-iface = gr.Interface(
-    fn=analyze_image,
-    inputs=[
-        gr.Image(type="pil", label="Upload an Image"),
-        gr.Textbox(label="Prompt", placeholder="Enter your prompt, e.g., 'Was siehst du auf dem Bild?' or 'List last 8 candles with their colors'")
-    ],
-    outputs="json",
-    title="Chart Analysis with YOLOv8 and EasyOCR",
-    description="Upload a TradingView screenshot to analyze candlesticks or get a general description."
-)
-
-iface.launch()
+        return {"prompt": prompt, "description": "Unsupported prompt. Use 'Was siehst du auf dem Bild?' or 'List last 8 candles with their colors'."