Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,15 +13,17 @@ def run(image, model_name):
|
|
| 13 |
file_path = "temp.png"
|
| 14 |
image.save(file_path)
|
| 15 |
print(file_path)
|
| 16 |
-
|
|
|
|
| 17 |
result = shap_e_client.predict(
|
| 18 |
file_path, # filepath in 'Input image' Image component
|
| 19 |
0, # float (numeric value between 0 and 2147483647) in 'Seed' Slider component
|
| 20 |
1, # float (numeric value between 1 and 20) in 'Guidance scale' Slider component
|
| 21 |
2, # float (numeric value between 2 and 100) in 'Number of inference steps' Slider component
|
| 22 |
api_name="/image-to-3d")
|
|
|
|
| 23 |
|
| 24 |
-
elif model_name=='
|
| 25 |
|
| 26 |
process_result = triposr_client.predict(
|
| 27 |
image, # filepath in 'Input Image' Image component
|
|
|
|
| 13 |
file_path = "temp.png"
|
| 14 |
image.save(file_path)
|
| 15 |
print(file_path)
|
| 16 |
+
model_name = model_name.lower()
|
| 17 |
+
if model_name=='shap-e':
|
| 18 |
result = shap_e_client.predict(
|
| 19 |
file_path, # filepath in 'Input image' Image component
|
| 20 |
0, # float (numeric value between 0 and 2147483647) in 'Seed' Slider component
|
| 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 |
|
| 28 |
process_result = triposr_client.predict(
|
| 29 |
image, # filepath in 'Input Image' Image component
|