Spaces:

Hrishikesh332
/

autotrain-meme-classification

Runtime error

Hrishikesh332 commited on Mar 24, 2023

Commit

326eb74

1 Parent(s): dbc4e8e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ if img is not None:
     st.write(img)
     extractor = AutoFeatureExtractor.from_pretrained("Hrishikesh332/autotrain-meme-classification-42897109437")
     model = AutoModelForImageClassification.from_pretrained("Hrishikesh332/autotrain-meme-classification-42897109437")
-    pred=model.predict(img)
     st.code(pred)

     st.write(img)
     extractor = AutoFeatureExtractor.from_pretrained("Hrishikesh332/autotrain-meme-classification-42897109437")
     model = AutoModelForImageClassification.from_pretrained("Hrishikesh332/autotrain-meme-classification-42897109437")
+    pred=model(img)
     st.code(pred)