AlexWortega commited on
Commit
01e1f7a
1 Parent(s): 66759a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ template = 'белков: '
34
  def classify_image(inp):
35
  print(type(inp))
36
  inp = Image.fromarray(inp)
37
- texts = generate_captions(inp, tokenizer, model, vae, template=template, limit_eos=False, top_k=16, captions_num=1, bs=16, top_p=0.6, seed=43, temperature=0.8)
38
  rp = texts[0].replace('белков','protein').replace('жиров','fat').replace('углеводов','carbs').replace('calories','ккал')
39
  print(rp)
40
 
 
34
  def classify_image(inp):
35
  print(type(inp))
36
  inp = Image.fromarray(inp)
37
+ texts = generate_captions(inp, tokenizer, model, vae, template=template, top_k=16, captions_num=1, bs=16, top_p=0.6, seed=43, temperature=0.8)
38
  rp = texts[0].replace('белков','protein').replace('жиров','fat').replace('углеводов','carbs').replace('calories','ккал')
39
  print(rp)
40