Omnibus commited on
Commit
69360fc
1 Parent(s): 94b598f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -16,9 +16,9 @@ models=[
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=[]
@@ -252,4 +252,4 @@ with gr.Blocks() as app:
252
  btn.click(image_classifier1,[inp],[n_out4]).then(tot_prob,None,fin,show_progress=False)
253
  btn.click(image_classifier2,[inp],[n_out5]).then(tot_prob,None,fin,show_progress=False)
254
 
255
- app.queue(concurrency_count=6).launch(show_api=False,max_threads=24)
 
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=[]
 
252
  btn.click(image_classifier1,[inp],[n_out4]).then(tot_prob,None,fin,show_progress=False)
253
  btn.click(image_classifier2,[inp],[n_out5]).then(tot_prob,None,fin,show_progress=False)
254
 
255
+ app.launch(show_api=False,max_threads=24)