Spaces:
Build error
Build error
AlexWortega
commited on
Commit
•
01e1f7a
1
Parent(s):
66759a7
Update app.py
Browse files
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,
|
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 |
|