Omnibus commited on
Commit
94b598f
·
1 Parent(s): f962b38

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -16,9 +16,9 @@ models=[
16
  "arnolfokam/ai-generated-image-detector",
17
  ]
18
 
19
- pipe0 = pipeline("image-classification", f"{models[0]}")
20
- pipe1 = pipeline("image-classification", f"{models[1]}")
21
- pipe2 = pipeline("image-classification", f"{models[2]}")
22
 
23
 
24
  fin_sum=[]
 
16
  "arnolfokam/ai-generated-image-detector",
17
  ]
18
 
19
+ pipe0 = pipeline("image-classification", f"{models[0]}",batch=True, max_batch_size=4)
20
+ pipe1 = pipeline("image-classification", f"{models[1]}",batch=True, max_batch_size=4)
21
+ pipe2 = pipeline("image-classification", f"{models[2]}",batch=True, max_batch_size=4)
22
 
23
 
24
  fin_sum=[]