Anonumous commited on
Commit
04fc647
1 Parent(s): d3ad28d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -212,7 +212,7 @@ def get_ans(clip_emb, prompt):
212
 
213
 
214
  device = 'cpu'
215
- clip_model, preprocess = clip.load("ViT-L-14-336px", device=device, jit=False)
216
  tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3medium_based_on_gpt2')
217
  prefix_length = 30
218
  model_path = 'prefix_small_latest_gpt2_medium.pt'
 
212
 
213
 
214
  device = 'cpu'
215
+ clip_model, preprocess = clip.load("ViT-L/14@336px", device=device, jit=False)
216
  tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3medium_based_on_gpt2')
217
  prefix_length = 30
218
  model_path = 'prefix_small_latest_gpt2_medium.pt'