Upload app.py

app.py

@@ -0,0 +1,110 @@
+import gradio as gr
+import torch
+from huggingface_hub import HfApi, login, whoami
+import os
+import sys
+import traceback
+from getpass import getpass
+
+# Force CPU usage if needed
+device = "cuda" if torch.cuda.is_available() else "cpu"
+print(f"Using device: {device}")
+
+# More details about the environment
+print(f"Gradio version: {gr.__version__}")
+print(f"Python version: {sys.version}")
+
+def validate_token(token):
+    """Validate a Hugging Face token by attempting to get user info."""
+    try:
+        login(token=token)
+        user_info = whoami()
+        print(f"Successfully logged in as: {user_info['name']} ({user_info.get('email', 'N/A')})")
+        return True
+    except Exception as e:
+        print(f"Token validation error: {str(e)}")
+        return False
+
+# Try to get token from environment variable
+hf_token = os.getenv("HUGGINGFACE_TOKEN")
+
+# If not found in environment variables, prompt for manual input
+if not hf_token:
+    print("HUGGINGFACE_TOKEN not found in environment variables.")
+    print("You can set it permanently with:")
+    print("  export HUGGINGFACE_TOKEN=your_token_here  # On Linux/macOS")
+    print("  set HUGGINGFACE_TOKEN=your_token_here     # On Windows")
+    print("\nAlternatively, you can enter your token manually below:")
+    
+    try:
+        hf_token = getpass("Enter your Hugging Face token (input will be hidden): ")
+    except Exception as e:
+        print(f"Error reading token: {str(e)}")
+        print("Continuing without token...")
+
+# Validate token if available
+if hf_token:
+    token_valid = validate_token(hf_token)
+    if not token_valid:
+        print("Warning: Token validation failed. Model loading may not work.")
+else:
+    print("Warning: No Hugging Face token provided. Model loading may not work.")
+
+def load_model():
+    try:
+        print("Attempting to load model from Hugging Face Hub...")
+        # Try direct Hub access first
+        try:
+            from huggingface_hub import snapshot_download
+            print("Trying to access model directory...")
+            model_path = snapshot_download(repo_id="goofyai/3d_render_style_xl", token=hf_token)
+            print(f"Model downloaded to {model_path}")
+        except Exception as e:
+            print(f"Error downloading model: {str(e)}")
+        
+        # Try different approaches to load the model
+        try:
+            print("Using gr.load() method with explicit token...")
+            interface = gr.load(
+                "models/goofyai/3d_render_style_xl",
+                hf_token=hf_token
+            )
+            return interface
+        except Exception as e:
+            print(f"Error with gr.load() using explicit token: {str(e)}")
+            print(traceback.format_exc())
+            
+            # Try without specifying token (using the login from earlier)
+            try:
+                print("Using gr.load() without explicit token...")
+                interface = gr.load("models/goofyai/3d_render_style_xl")
+                return interface
+            except Exception as e:
+                print(f"Error with gr.load() without token: {str(e)}")
+                print(traceback.format_exc())
+                
+                # Try old Interface.load as last resort
+                print("Trying Interface.load() as fallback...")
+                interface = gr.Interface.load("models/goofyai/3d_render_style_xl")
+                return interface
+    except Exception as e:
+        print(f"Error loading model: {str(e)}")
+        print(traceback.format_exc())
+        return None
+
+# Create the interface
+try:
+    interface = load_model()
+    if interface:
+        print("Model loaded successfully, launching interface...")
+        interface.launch(
+            share=False,
+            server_name="0.0.0.0",
+            server_port=7860,
+            show_error=True
+        )
+    else:
+        print("Failed to load the interface")
+except Exception as e:
+    print(f"Error launching interface: {str(e)}")
+    print(traceback.format_exc())