yejunliang23 commited on
Commit
ba328c3
·
verified ·
1 Parent(s): 2a61ab0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -267,7 +267,11 @@ def add_file(history, task_history, file, task_new, fig, query):
267
  token = vqvae.Encode(ss.to(dtype=torch.float32).unsqueeze(0))
268
  token = token[0].cpu().numpy().tolist()
269
  words = token_to_words(token)
270
- fig = make_pointcloud_figure(position_recon,rotate=True)
 
 
 
 
271
  return history, task_history,file.name,task_new,fig,gr.update(
272
  value=f"{words}\nGive a quick overview of the object represented by this 3D mesh.")
273
  history = history if history is not None else []
 
267
  token = vqvae.Encode(ss.to(dtype=torch.float32).unsqueeze(0))
268
  token = token[0].cpu().numpy().tolist()
269
  words = token_to_words(token)
270
+ print(words)
271
+ print(position_recon.shape)
272
+ print(position_recon.max())
273
+ print(position_recon.min())
274
+ fig = make_pointcloud_figure(position_recon,rotate=True)
275
  return history, task_history,file.name,task_new,fig,gr.update(
276
  value=f"{words}\nGive a quick overview of the object represented by this 3D mesh.")
277
  history = history if history is not None else []