Geonmo commited on
Commit
be8777a
1 Parent(s): 8d32bca

clean all codes

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -105,7 +105,6 @@ def load_models(args):
105
  @torch.no_grad()
106
  def predict(images, input_text, model_name):
107
  start_time = time.time()
108
- print(images)
109
  input_images = model_dict['clip_preprocess'](images, return_tensors='pt')['pixel_values'].to(model_dict['device'])
110
  input_text = input_text.replace('$', '[$]')
111
  input_tokens = model_dict['tokenizer'](text=input_text, return_tensors='pt', padding='max_length', truncation=True)['input_ids'].to(model_dict['device'])
 
105
  @torch.no_grad()
106
  def predict(images, input_text, model_name):
107
  start_time = time.time()
 
108
  input_images = model_dict['clip_preprocess'](images, return_tensors='pt')['pixel_values'].to(model_dict['device'])
109
  input_text = input_text.replace('$', '[$]')
110
  input_tokens = model_dict['tokenizer'](text=input_text, return_tensors='pt', padding='max_length', truncation=True)['input_ids'].to(model_dict['device'])