Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -64,8 +64,7 @@ def save_ply_from_array(verts):
|
|
| 64 |
tmpf.flush(); tmpf.close()
|
| 65 |
return tmpf.name
|
| 66 |
|
| 67 |
-
|
| 68 |
-
@spaces.GPU
|
| 69 |
def predict(_chatbot,task_history,viewer_voxel,viewer_mesh,task_new,seed,top_k,top_p,temperature):
|
| 70 |
torch.manual_seed(seed)
|
| 71 |
chat_query = _chatbot[-1][0]
|
|
@@ -257,6 +256,7 @@ def add_text(history, task_history, text,task_new):
|
|
| 257 |
task_new = task_new + [(task_text, None)]
|
| 258 |
return history, task_history,task_new
|
| 259 |
|
|
|
|
| 260 |
def add_file(history, task_history, file, task_new, fig, query):
|
| 261 |
if file.name.endswith(('.obj', '.glb')):
|
| 262 |
position_recon = load_vertices(file.name)#(N,3)
|
|
|
|
| 64 |
tmpf.flush(); tmpf.close()
|
| 65 |
return tmpf.name
|
| 66 |
|
| 67 |
+
@spaces.GPU(duration=120)
|
|
|
|
| 68 |
def predict(_chatbot,task_history,viewer_voxel,viewer_mesh,task_new,seed,top_k,top_p,temperature):
|
| 69 |
torch.manual_seed(seed)
|
| 70 |
chat_query = _chatbot[-1][0]
|
|
|
|
| 256 |
task_new = task_new + [(task_text, None)]
|
| 257 |
return history, task_history,task_new
|
| 258 |
|
| 259 |
+
@spaces.GPU(duration=120)
|
| 260 |
def add_file(history, task_history, file, task_new, fig, query):
|
| 261 |
if file.name.endswith(('.obj', '.glb')):
|
| 262 |
position_recon = load_vertices(file.name)#(N,3)
|