yejunliang23 commited on
Commit
cc303c8
·
verified ·
1 Parent(s): 3f604b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -60,6 +60,7 @@ def save_ply_from_array(verts):
60
  tmpf.flush(); tmpf.close()
61
  return tmpf.name
62
 
 
63
  def predict(_chatbot,task_history,viewer_voxel,viewer_mesh,task_new,seed,top_k,top_p,temperature):
64
  torch.manual_seed(seed)
65
  chat_query = _chatbot[-1][0]
 
60
  tmpf.flush(); tmpf.close()
61
  return tmpf.name
62
 
63
+ @spaces.GPU(duration=120)
64
  def predict(_chatbot,task_history,viewer_voxel,viewer_mesh,task_new,seed,top_k,top_p,temperature):
65
  torch.manual_seed(seed)
66
  chat_query = _chatbot[-1][0]