ZhiyuanthePony commited on
Commit
9b177de
·
1 Parent(s): cca7700
Files changed (1) hide show
  1. app.py +3 -2
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