LPX55 commited on
Commit
05893f9
·
verified ·
1 Parent(s): 4a03e59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,8 +26,8 @@ clf_2 = pipeline("image-classification", model=model_2_path)
26
 
27
  # Load additional models
28
  models = ["Organika/sdxl-detector", "cmckinle/sdxl-flux-detector"]
29
- pipe0 = pipeline("image-classification", model=models[0])
30
- pipe1 = pipeline("image-classification", model=models[1])
31
 
32
  # Define class names for all models
33
  class_names_1 = ['artificial', 'real']
 
26
 
27
  # Load additional models
28
  models = ["Organika/sdxl-detector", "cmckinle/sdxl-flux-detector"]
29
+ pipe0 = pipeline("image-classification", model=models[0], device=device)
30
+ pipe1 = pipeline("image-classification", model=models[1], device=device)
31
 
32
  # Define class names for all models
33
  class_names_1 = ['artificial', 'real']