text-to-3dmodel / app.py
itriedcoding's picture
Deploy custom model via Nexus AI Studio
3724391 verified
raw
history blame contribute delete
745 Bytes
import gradio as gr
from huggingface_hub import InferenceClient
model_id = "stabilityai/stable-zero123"
client = InferenceClient(model_id)
def generate_3d(prompt):
try:
response = client.post(json={"inputs": prompt})
filename = "output.glb"
with open(filename, "wb") as f:
f.write(response)
return filename
except Exception as e:
raise gr.Error(str(e))
demo = gr.Interface(
fn=generate_3d,
inputs=gr.Textbox(label="Prompt", placeholder="Describe the 3D model you want..."),
outputs=gr.Model3D(label="3D Model"),
title="text-to-3dmodel",
description="Text to 3D generation powered by " + model_id,
theme="soft"
)
if __name__ == "__main__":
demo.launch()