Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -88,7 +88,7 @@ def aiornot0(image):
|
|
88 |
fin_sum.append(results)
|
89 |
return gr.HTML.update(html_out),results
|
90 |
def aiornot1(image):
|
91 |
-
labels = ["
|
92 |
mod=models[1]
|
93 |
feature_extractor1 = AutoFeatureExtractor.from_pretrained(mod)
|
94 |
model1 = AutoModelForImageClassification.from_pretrained(mod)
|
@@ -113,7 +113,7 @@ def aiornot1(image):
|
|
113 |
fin_sum.append(results)
|
114 |
return gr.HTML.update(html_out),results
|
115 |
def aiornot2(image):
|
116 |
-
labels = ["
|
117 |
mod=models[2]
|
118 |
feature_extractor2 = AutoFeatureExtractor.from_pretrained(mod)
|
119 |
model2 = AutoModelForImageClassification.from_pretrained(mod)
|
|
|
88 |
fin_sum.append(results)
|
89 |
return gr.HTML.update(html_out),results
|
90 |
def aiornot1(image):
|
91 |
+
labels = ["AI", "Real"]
|
92 |
mod=models[1]
|
93 |
feature_extractor1 = AutoFeatureExtractor.from_pretrained(mod)
|
94 |
model1 = AutoModelForImageClassification.from_pretrained(mod)
|
|
|
113 |
fin_sum.append(results)
|
114 |
return gr.HTML.update(html_out),results
|
115 |
def aiornot2(image):
|
116 |
+
labels = ["Real", "AI"]
|
117 |
mod=models[2]
|
118 |
feature_extractor2 = AutoFeatureExtractor.from_pretrained(mod)
|
119 |
model2 = AutoModelForImageClassification.from_pretrained(mod)
|