AI-or-Not-dev

Runtime error

Omnibus commited on Oct 14, 2023

Commit

c7c6c17

1 Parent(s): f0307a8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -113,7 +113,7 @@ def aiornot1(image):
     fin_sum.append(results)
     return gr.HTML.update(html_out),results
 def aiornot2(image):
-    labels = ["Real", "AI"]
     mod=models[2]
     feature_extractor2 = AutoFeatureExtractor.from_pretrained(mod)
     model2 = AutoModelForImageClassification.from_pretrained(mod)

     fin_sum.append(results)
     return gr.HTML.update(html_out),results
 def aiornot2(image):
+    labels = ["AI", "Real"]
     mod=models[2]
     feature_extractor2 = AutoFeatureExtractor.from_pretrained(mod)
     model2 = AutoModelForImageClassification.from_pretrained(mod)