Omnibus commited on
Commit
f0307a8
1 Parent(s): 2ad5c84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = ["Real", "AI"]
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 = ["AI", "Real"]
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)