|
import gradio as gr |
|
import torch |
|
import os |
|
import sys |
|
from huggingface_hub import login |
|
|
|
|
|
device = "cuda" if torch.cuda.is_available() else "cpu" |
|
print(f"Using device: {device}") |
|
|
|
|
|
print(f"Gradio version: {gr.__version__}") |
|
print(f"Python version: {sys.version}") |
|
|
|
|
|
|
|
hf_token = os.environ.get("HUGGINGFACE_TOKEN") |
|
if hf_token: |
|
print("Found HUGGINGFACE_TOKEN in environment variables") |
|
|
|
login(token=hf_token) |
|
print("Logged in with Hugging Face token") |
|
else: |
|
print("HUGGINGFACE_TOKEN not found in environment variables") |
|
|
|
|
|
|
|
def load_model(): |
|
try: |
|
print("Attempting to load 3D render style model...") |
|
|
|
|
|
|
|
try: |
|
print("Trying to load from 'models' source...") |
|
interface = gr.load( |
|
name="goofyai/3d_render_style_xl", |
|
src="models" |
|
) |
|
return interface |
|
except Exception as e: |
|
print(f"Error loading from 'models': {str(e)}") |
|
|
|
|
|
try: |
|
print("Trying to load from 'huggingface' source...") |
|
interface = gr.load( |
|
name="goofyai/3d_render_style_xl", |
|
src="huggingface" |
|
) |
|
return interface |
|
except Exception as e: |
|
print(f"Error loading from 'huggingface': {str(e)}") |
|
|
|
|
|
print("Trying to load from 'spaces' source...") |
|
try: |
|
interface = gr.load( |
|
name="goofyai/3d_render_style_xl", |
|
src="spaces" |
|
) |
|
return interface |
|
except Exception as e: |
|
print(f"Error loading from 'spaces': {str(e)}") |
|
|
|
|
|
print("Trying legacy Interface.load()...") |
|
try: |
|
interface = gr.Interface.load( |
|
"models/goofyai/3d_render_style_xl" |
|
) |
|
return interface |
|
except Exception as e: |
|
print(f"Error with Interface.load(): {str(e)}") |
|
|
|
|
|
print("Attempting to create a basic model interface...") |
|
|
|
def generate_3d_render(prompt): |
|
|
|
|
|
return "Model yüklenemedi. Lütfen HuggingFace izinlerinizi kontrol edin." |
|
|
|
|
|
interface = gr.Interface( |
|
fn=generate_3d_render, |
|
inputs=gr.Textbox(label="Prompt", placeholder="3D render description"), |
|
outputs=gr.Textbox(label="Result"), |
|
title="3D Render Style XL", |
|
description="Enter a prompt to generate a 3D render" |
|
) |
|
|
|
return interface |
|
|
|
except Exception as e: |
|
print(f"Error loading model: {str(e)}") |
|
return None |
|
|
|
|
|
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)}") |