LPX commited on
Commit
737fdca
·
1 Parent(s): 7cd1bf7

feat: enhance predict function with detailed docstring and add queue to demo interface

Browse files
Files changed (1) hide show
  1. app.py +17 -0
app.py CHANGED
@@ -532,6 +532,22 @@ detection_model_eval_playground = gr.Interface(
532
 
533
 
534
  def predict(img):
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
535
  client = Client("aiwithoutborders-xyz/OpenSight-Community-Forensics-Preview")
536
  client.view_api()
537
  result = client.predict(
@@ -697,6 +713,7 @@ demo = gr.TabbedInterface(
697
 
698
  # Add live update for the logs textbox
699
  # demo.load(get_captured_logs, None, demo.outputs[8], every=1)
 
700
 
701
  if __name__ == "__main__":
702
  demo.launch(mcp_server=True)
 
532
 
533
 
534
  def predict(img):
535
+ """
536
+ Predicts whether an image is AI-generated or real using the SOTA Community Forensics model.
537
+
538
+ Args:
539
+ img (str): Path to the input image file to analyze.
540
+
541
+ Returns:
542
+ dict: A dictionary containing:
543
+ - 'Fake Probability' (float): Probability score between 0 and 1 indicating likelihood of being AI-generated
544
+ - 'Result Description' (str): Human-readable description of the prediction result
545
+
546
+ Example:
547
+ >>> result = predict("path/to/image.jpg")
548
+ >>> print(result)
549
+ {'Fake Probability': 0.002, 'Result Description': 'The image is likely real.'}
550
+ """
551
  client = Client("aiwithoutborders-xyz/OpenSight-Community-Forensics-Preview")
552
  client.view_api()
553
  result = client.predict(
 
713
 
714
  # Add live update for the logs textbox
715
  # demo.load(get_captured_logs, None, demo.outputs[8], every=1)
716
+ demo.queue(max_size=4)
717
 
718
  if __name__ == "__main__":
719
  demo.launch(mcp_server=True)