app.py

import gradio as gr
import torch
from huggingface_hub import HfApi, login
import os

# Force CPU usage if needed
device = "cuda" if torch.cuda.is_available() else "cpu"
print(f"Using device: {device}")

# Login to Hugging Face with token
hf_token = os.getenv("HUGGINGFACE_TOKEN")
if hf_token:
    login(token=hf_token)
else:
    print("Warning: HUGGINGFACE_TOKEN environment variable not set")

def load_model():
    try:
        interface = gr.Interface.load(
            "models/goofyai/3d_render_style_xl",
            alias="game-icon-generator"
        )
        return interface
    except Exception as e:
        print(f"Error loading model: {str(e)}")
        return None

# Create the interface
try:
    interface = load_model()
    if 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)}")