Spaces:

javierrf91
/

streamlit

Runtime error

javierrf91 commited on Mar 9, 2023

Commit

bdd33dd

1 Parent(s): e47a06b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,4 +22,4 @@ if file_name is not None:
     input_pixels = inputs.pixel_values
     model = AutoModelForImageClassification.from_pretrained('javierrf91/streamlit')
     outputs = model(input_pixels)
-    col2.header("Predicted class:" + model.config.id2label[outputs.logits.argmax(-1)  .item()])

     input_pixels = inputs.pixel_values
     model = AutoModelForImageClassification.from_pretrained('javierrf91/streamlit')
     outputs = model(input_pixels)
+    col2.header(model.config.id2label[outputs.logits.argmax(-1)  .item()])