AI-or-Not-dev

Runtime error

Omnibus commited on Oct 13, 2023

Commit

391de58

1 Parent(s): 6fc4472

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -58,7 +58,7 @@ def aiornot1(image):
     AI: {px[1][0]}"""
     return gr.HTML.update(html_out)
 def aiornot2(image):
-    labels = ["Real", "AI"]
     mod=models[2]
     feature_extractor2 = AutoFeatureExtractor.from_pretrained(mod)
     model2 = AutoModelForImageClassification.from_pretrained(mod)

     AI: {px[1][0]}"""
     return gr.HTML.update(html_out)
 def aiornot2(image):
+    labels = ["AI", "Real"]
     mod=models[2]
     feature_extractor2 = AutoFeatureExtractor.from_pretrained(mod)
     model2 = AutoModelForImageClassification.from_pretrained(mod)