dnm3d / app.py
geyik1's picture
Upload app.py
8dad306 verified
import gradio as gr
import torch
import os
import sys
from huggingface_hub import login
# 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}")
# Hugging Face API token'ı - önce environment variable olarak ara,
# sonra Hugging Face Secrets sisteminde ara
hf_token = os.environ.get("HUGGINGFACE_TOKEN")
if hf_token:
print("Found HUGGINGFACE_TOKEN in environment variables")
# Token ile giriş yap
login(token=hf_token)
print("Logged in with Hugging Face token")
else:
print("HUGGINGFACE_TOKEN not found in environment variables")
# Hugging Face Spaces bu değişkeni otomatik olarak yükleyecek
# eğer Spaces UI üzerinden secret olarak eklediyseniz
def load_model():
try:
print("Attempting to load 3D render style model...")
# Gradio 5.29.1 sürümü için doğru format:
# gr.load() fonksiyonu için 'src' parametresi belirtilmeli
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)}")
# Hugging Face kaynağını deneyelim
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)}")
# Space kaynağını deneyelim
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)}")
# Son çare: Interface.load() deneyelim
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)}")
# En son çare: doğrudan 3D model oluşturma yaklaşımı
print("Attempting to create a basic model interface...")
def generate_3d_render(prompt):
# Burada gerçek model çağrısı olmadan basit bir arayüz dönüyoruz
# Asıl amacımız arayüzün çalışmasını sağlamak
return "Model yüklenemedi. Lütfen HuggingFace izinlerinizi kontrol edin."
# Basit bir Gradio arayüzü oluştur
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
# 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)}")