Update app.py

app.py

@@ -32,14 +32,14 @@ allowed_tags.append("explicit")
 allowed_tags.append("questionable")
 allowed_tags.append("safe")
 
-def create_tags(image):
+def create_tags(image, thres):
   img = image.convert('RGB')
   tensor = transform(img).unsqueeze(0)
 
   with torch.no_grad():
       out = model(tensor)
       probabilities = torch.nn.functional.sigmoid(out[0])
-      indices = torch.where(probabilities > 0.3)[0]
+      indices = torch.where(probabilities > thres)[0]
       values = probabilities[indices]
 
   temp = []
@@ -56,7 +56,7 @@ def create_tags(image):
 
 demo = gr.Interface(
     fn=create_tags,
-    inputs=[gr.Image(type="pil")],
+    inputs=[gr.Image(type="pil"), gr.Slider(minimum=0, maximum = 1, step = 0.01, value = 0.3)],
     outputs=["text"],
 )
-demo.launch(debug = True)
+demo.launch(debug = True)