yejunliang23 commited on
Commit
fd9c629
·
verified ·
1 Parent(s): d3f06b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -64,8 +64,7 @@ def save_ply_from_array(verts):
64
  tmpf.flush(); tmpf.close()
65
  return tmpf.name
66
 
67
- #@spaces.GPU(duration=120)
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)