Update app.py

app.py

@@ -38,9 +38,9 @@ def save_media_and_get_path(media, media_type):
 
 @spaces.GPU
 def qwen_inference(media, text_input=None):
-    if isinstance(media, Image.Image):
+    if media.endswith(Image.registered_extensions()):
         media_type = "image"
-    elif isinstance(media, str) and media.endswith((".mp4", ".webm", ".avi")):  # Check if it's a video path
+    elif media.endswith(("avi", "mp4", "mov", "mkv", "flv", "wmv", "mjpeg", "wav", "gif", "webm", "m4v", "3gp")):  # Check if it's a video path
         media_type = "video"
     else:
         raise ValueError("Unsupported media type. Please upload an image or video.")
@@ -54,7 +54,7 @@ def qwen_inference(media, text_input=None):
                 {
                     "type": media_type,
                     media_type: media_path,
-                    **({"max_pixels": 360 * 420, "fps": 6.0} if media_type == "video" else {}), 
+                    **({"max_pixels": 500 * 500, "fps": 8.0} if media_type == "video" else {}), 
                 },
                 {"type": "text", "text": text_input},
             ],