Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
9b177de
1
Parent(s):
cca7700
update
Browse files
app.py
CHANGED
@@ -5,6 +5,7 @@ try:
|
|
5 |
import spaces
|
6 |
except:
|
7 |
pass
|
|
|
8 |
|
9 |
# Check if setup has been run
|
10 |
setup_marker = ".setup_complete"
|
@@ -22,7 +23,7 @@ if not os.path.exists(setup_marker):
|
|
22 |
|
23 |
import torch
|
24 |
import gradio as gr
|
25 |
-
from typing import
|
26 |
from collections import deque
|
27 |
from diffusers import StableDiffusionPipeline
|
28 |
|
@@ -57,7 +58,7 @@ def initialize_pipeline():
|
|
57 |
return PIPELINE
|
58 |
|
59 |
@spaces.GPU
|
60 |
-
def generate_3d_mesh(prompt: str) -> Tuple[str, str]:
|
61 |
"""Generate 3D mesh from text prompt"""
|
62 |
global PIPELINE, OBJ_FILE_QUEUE
|
63 |
|
|
|
5 |
import spaces
|
6 |
except:
|
7 |
pass
|
8 |
+
os.environ["PYDANTIC_STRICT_TYPE_CHECKING"] = "0"
|
9 |
|
10 |
# Check if setup has been run
|
11 |
setup_marker = ".setup_complete"
|
|
|
23 |
|
24 |
import torch
|
25 |
import gradio as gr
|
26 |
+
from typing import Tuple, List, Dict, Any, Optional
|
27 |
from collections import deque
|
28 |
from diffusers import StableDiffusionPipeline
|
29 |
|
|
|
58 |
return PIPELINE
|
59 |
|
60 |
@spaces.GPU
|
61 |
+
def generate_3d_mesh(prompt: str) -> Tuple[Optional[str], Optional[str]]:
|
62 |
"""Generate 3D mesh from text prompt"""
|
63 |
global PIPELINE, OBJ_FILE_QUEUE
|
64 |
|