Update app.py

app.py

@@ -5,7 +5,7 @@ import PIL.Image
 
 from gradio_client import Client
 
-shap_e_client = Client("hysts/Shap-E")
+lgm_mini_client = Client("dylanebert/LGM-mini")
 triposr_client = Client("stabilityai/TripoSR")
 
 
@@ -14,13 +14,11 @@ def run(image, model_name):
     image.save(file_path)
     print(file_path)
     model_name = model_name.lower()
-    if model_name=='shap-e':
-        result = shap_e_client.predict(
-		file_path,	# filepath  in 'Input image' Image component
-		0,	# float (numeric value between 0 and 2147483647) in 'Seed' Slider component
-		1,	# float (numeric value between 1 and 20) in 'Guidance scale' Slider component
-		2,	# float (numeric value between 2 and 100) in 'Number of inference steps' Slider component
-		api_name="/image-to-3d")
+    if model_name=='lgm-mini':
+        result = lgm_mini_client.predict(
+        		file_path,	# filepath  in 'image' Image component
+        		api_name="/run"
+        )
         output = result
 
     elif model_name=='triposr':
@@ -37,7 +35,7 @@ def run(image, model_name):
 		32,	# float (numeric value between 32 and 320) in 'Marching Cubes Resolution' Slider component
 		api_name="/generate")
 
-        output = result[1]
+        output = result[0]
             
     return output