Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=[]
|