Spaces:
Runtime error
Runtime error
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=[]
|