LPX55 commited on
Commit
f8c9444
·
verified ·
1 Parent(s): 09e67fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ model_3 = AutoModelForImageClassification.from_pretrained(models[0]).to(device)
36
  feature_extractor_4 = AutoFeatureExtractor.from_pretrained(models[1], device=device)
37
  model_4 = AutoModelForImageClassification.from_pretrained(models[1]).to(device)
38
 
39
- model_5 = load_model("large_model_3lakh_v1.h5", device=device)
40
  # Define class names for all models
41
  class_names_1 = ['artificial', 'real']
42
  class_names_2 = ['AI Image', 'Real Image']
 
36
  feature_extractor_4 = AutoFeatureExtractor.from_pretrained(models[1], device=device)
37
  model_4 = AutoModelForImageClassification.from_pretrained(models[1]).to(device)
38
 
39
+ model_5 = load_model("large_model_3lakh_v1.h5")
40
  # Define class names for all models
41
  class_names_1 = ['artificial', 'real']
42
  class_names_2 = ['AI Image', 'Real Image']