NikhilJoson commited on
Commit
446a044
·
verified ·
1 Parent(s): e663763

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -34,7 +34,8 @@ def predict_clothing(images):
34
  with torch.no_grad():
35
  output = model.generate(**inputs, max_new_tokens=32)
36
 
37
- output_texts.append(str(processor.decode(output[0])).split('\n').[-1])
 
38
 
39
  print(output_texts)
40
  return output_texts
 
34
  with torch.no_grad():
35
  output = model.generate(**inputs, max_new_tokens=32)
36
 
37
+ output_reponse = str(processor.decode(output[0])).split('\n')
38
+ output_texts.append(output_reponse[-1])
39
 
40
  print(output_texts)
41
  return output_texts