Spaces:

bharathj16
/

Test1

Sleeping

bharathj16 commited on Nov 19, 2023

Commit

a655929

•

1 Parent(s): 99947f3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,4 +13,4 @@ outputs = model(**inputs)
 logits = outputs.logits
 # model predicts one of the 1000 ImageNet classes
 predicted_class_idx = logits.argmax(-1).item()
-print("Predicted class:", model.config.id2label[predicted_class_idx])

 logits = outputs.logits
 # model predicts one of the 1000 ImageNet classes
 predicted_class_idx = logits.argmax(-1).item()
+print("Predicted class:", model.config.id2label[predicted_class_idx])