vumichien commited on
Commit
fe0189b
1 Parent(s): c20195c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -230,6 +230,7 @@ def render(motions, device_id=0, name='test_vis'):
230
 
231
  def predict(clip_text, method='fast'):
232
  gc.collect()
 
233
  if torch.cuda.is_available():
234
  text = clip.tokenize([clip_text], truncate=True).cuda()
235
  else:
 
230
 
231
  def predict(clip_text, method='fast'):
232
  gc.collect()
233
+ print('prompt text instruction: {}'.format(clip_text))
234
  if torch.cuda.is_available():
235
  text = clip.tokenize([clip_text], truncate=True).cuda()
236
  else: