Update app.py

app.py

@@ -157,8 +157,11 @@ class S3DirectStream:
                 if file_info.rfilename.endswith(('.bin', '.safetensors', 'config.json', 'tokenizer.json')):
                     file_url = api.download_file(model_prefix, file_info.rfilename)
                     s3_key = f"{model_prefix}/{file_info.rfilename}"
-                    self.download_and_upload_to_s3_url(file_url, s3_key)
-                    logger.info(f"Downloaded and uploaded {s3_key}")
+                    try:
+                        self.download_and_upload_to_s3_url(file_url, s3_key)
+                        logger.info(f"Downloaded and uploaded {s3_key}")
+                    except Exception as e:
+                        logger.exception(f"Error downloading/uploading {s3_key}: {e}")
 
             logger.info(f"Finished downloading and uploading model files for {model_prefix}.")
 
@@ -171,7 +174,7 @@ class S3DirectStream:
 
 
     def _get_model_files(self, model_prefix, revision="main"):
-        index_url = f"https://huggingface.co/{model_prefix}/tree/{revision}/"
+        index_url = f"https://huggingface.co/{model_prefix}/resolve/{revision}/"
         try:
             index_response = requests.get(index_url)
             index_response.raise_for_status()