Update app.py

app.py

@@ -208,16 +208,24 @@ class YOLOWorldDetector:
         # Run inference based on model type
         if self.model_type == "yoloworld":
             try:
+                # Parse text prompt properly for YOLOWorld
+                if text_prompt and text_prompt.strip():
+                    # Split by comma and strip whitespace
+                    classes = [cls.strip() for cls in text_prompt.split(',') if cls.strip()]
+                else:
+                    classes = None
+                
                 # YOLOWorld supports text prompts
                 results = self.model.predict(
                     source=image,
-                    classes=text_prompt.split(','),
+                    classes=classes,  # Pass as a list of strings or None
                     conf=confidence_threshold,
                     verbose=False
                 )
             except Exception as e:
                 print(f"Error during YOLOWorld inference: {e}")
-                # If YOLOWorld inference fails, try to use it as standard YOLO
+                print("Falling back to standard YOLO inference...")
+                # If YOLOWorld inference fails, use standard YOLO
                 results = self.model.predict(
                     source=image,
                     conf=confidence_threshold,
@@ -449,4 +457,4 @@ with gr.Blocks(title="YOLO Vision Suite", css=custom_css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=True)  # Set share=True to create a public link