Update app.py

app.py

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