Update app.py

app.py

@@ -209,7 +209,13 @@ class S3DirectStream:
         try:
             api = HfApi()
             model_info = api.model_info(model_prefix)
-            return model_info.default_revision
+            revision = model_info.revision
+            if revision:
+                return revision
+            else:
+                logger.warning(f"No default revision found for {model_prefix}, using 'main'")
+                return "main"
+
         except Exception as e:
             logger.error(f"Error getting latest revision for {model_prefix}: {e}")
             return None