Update app.py

app.py

@@ -33,7 +33,7 @@ def attempt_download_from_hub(repo_id, hf_token=None):
     from huggingface_hub.utils._validators import HFValidationError
     try:
         repo_files = list_repo_files(repo_id=repo_id, repo_type='model', token=hf_token)
-        model_file = [f for f in repo_files if f.endswith('.pt')][0]
+        model_file = [f for f in repo_files if f.endswith('.safetensors')][0]
         file = hf_hub_download(
             repo_id=repo_id,
             filename=model_file,