RamAnanth1 commited on
Commit
80ea7a7
1 Parent(s): c58857c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -21,7 +21,7 @@ def run(image, model_name):
21
  1, # float (numeric value between 1 and 20) in 'Guidance scale' Slider component
22
  2, # float (numeric value between 2 and 100) in 'Number of inference steps' Slider component
23
  api_name="/image-to-3d")
24
- print(result)
25
 
26
  elif model_name=='triposr':
27
 
@@ -37,8 +37,9 @@ def run(image, model_name):
37
  32, # float (numeric value between 32 and 320) in 'Marching Cubes Resolution' Slider component
38
  api_name="/generate")
39
 
40
-
41
- return result
 
42
 
43
 
44
  with gr.Blocks() as demo:
 
21
  1, # float (numeric value between 1 and 20) in 'Guidance scale' Slider component
22
  2, # float (numeric value between 2 and 100) in 'Number of inference steps' Slider component
23
  api_name="/image-to-3d")
24
+ output = result
25
 
26
  elif model_name=='triposr':
27
 
 
37
  32, # float (numeric value between 32 and 320) in 'Marching Cubes Resolution' Slider component
38
  api_name="/generate")
39
 
40
+ output = result[1]
41
+
42
+ return output
43
 
44
 
45
  with gr.Blocks() as demo: