space update

app.py

@@ -168,8 +168,20 @@ def load_model_from_file(uploaded_file):
         return "❌ No file uploaded"
     
     try:
+        # Check if the file path exists and is valid
+        file_path = uploaded_file.name if hasattr(uploaded_file, 'name') else str(uploaded_file)
+        
+        # For HF Spaces compatibility, also try the upload path
+        if not os.path.exists(file_path) and hasattr(uploaded_file, 'orig_name'):
+            # Sometimes HF Spaces provides different paths
+            print(f"Original path not found: {file_path}")
+            print(f"Trying original name: {uploaded_file.orig_name}")
+            file_path = uploaded_file.orig_name
+        
+        print(f"📁 Attempting to load model from: {file_path}")
+        
         # Load the new model
-        new_model, new_tokenizer, status = load_model(uploaded_file.name)
+        new_model, new_tokenizer, status = load_model(file_path)
         
         # Update global variables
         model = new_model
@@ -179,6 +191,7 @@ def load_model_from_file(uploaded_file):
         return status
     except Exception as e:
         error_msg = f"❌ Failed to load uploaded model: {e}"
+        print(f"Error details: {e}")
         model_status = error_msg
         return error_msg
 
@@ -249,7 +262,7 @@ with gr.Blocks(title="🤖 Usta Model Chat") as demo:
         load_url_btn = gr.Button("Load from URL", scale=1)
     
     with gr.Row():
-        model_file = gr.File(label="Upload .pth file", file_types=[".pth"])
+        model_file = gr.File(label="Upload model file (.pth, .pt, .bin)")
         load_file_btn = gr.Button("Load File", scale=1)
     
     status = gr.Textbox(label="Status", value=model_status, interactive=False)