Upload app.py

app.py

@@ -0,0 +1,69 @@
+import gradio as gr
+import torch
+from huggingface_hub import HfApi, login
+import os
+import sys
+import traceback
+
+# 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}")
+
+# Login to Hugging Face with token
+hf_token = os.getenv("HUGGINGFACE_TOKEN")
+if hf_token:
+    login(token=hf_token)
+    print("Successfully logged in to Hugging Face Hub")
+else:
+    print("Warning: HUGGINGFACE_TOKEN environment variable not set")
+
+def load_model():
+    try:
+        print("Attempting to load model from Hugging Face Hub...")
+        # Attempt to use HfApi to check if the model exists
+        api = HfApi()
+        try:
+            # Check if model exists
+            model_info = api.model_info("goofyai/3d_render_style_xl")
+            print(f"Found model: {model_info.modelId}")
+        except Exception as e:
+            print(f"Error checking model info: {str(e)}")
+        
+        # Try a different approach with the newer Gradio API
+        try:
+            print("Using gr.load() method...")
+            interface = gr.load("models/goofyai/3d_render_style_xl")
+            return interface
+        except Exception as e:
+            print(f"Error with gr.load(): {str(e)}")
+            print(traceback.format_exc())
+            
+            # Try the older Interface.load method as fallback
+            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())