Update app.py

app.py

@@ -54,6 +54,8 @@ def classify_text(inp):
     logits = torch.nn.Softmax(dim=1)(logits).squeeze(0).cpu().numpy()
     img_idx = match_img(logits)
     
+    img_output = 
+    
     # print(logits)
     return {labels[i]: float(logits[i].item()) for i in range(len(labels))}
     # return {logits.argmax().item() : labels[logits.argmax().item()]}