NikhilJoson commited on
Commit
e663763
·
verified ·
1 Parent(s): 63ac632

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -33,7 +33,8 @@ def predict_clothing(images):
33
  inputs = processor(image, input_text, add_special_tokens=False, return_tensors="pt").to(model.device)
34
  with torch.no_grad():
35
  output = model.generate(**inputs, max_new_tokens=32)
36
- output_texts.append(str(processor.decode(output[0])))
 
37
 
38
  print(output_texts)
39
  return output_texts
 
33
  inputs = processor(image, input_text, add_special_tokens=False, return_tensors="pt").to(model.device)
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