display title

app.py

@@ -1,11 +1,15 @@
 import gradio as gr
 
-from ppl_detector import detect
+from ppl_detector import detect, set_result
 
 with gr.Blocks() as demo:
-    img_input = gr.Image(type="filepath", label="Input Image")
+    img_input = gr.Image(type="filepath", format="jpg", label="Input Image")
+    
+    title = "<h1 id='result' style='display: flex;justify-content: center;'>it counts ppl, upload an image, see how it goes</h1>"
+    state = gr.HTML(title)
     img_output = gr.Image(label="Output Image with Detections")
     
     img_input.change(fn=detect, inputs=img_input, outputs=img_output)
+    img_output.change(fn=set_result, outputs=state)
        
 demo.launch()

ppl_detector.py

@@ -1,9 +1,10 @@
 from transformers import pipeline
-
 from PIL import Image, ImageDraw
+from pprint import pprint
 
 detector = pipeline("object-detection", model="hustvl/yolos-base")
 source = None
+state = None
 
 def draw_bbox(bboxes):
     global source
@@ -15,24 +16,34 @@ def draw_bbox(bboxes):
 
 
 def detect(img):
-    global source
-    if source is None and img is not None:
+    global source, result
+    if img is not None:
         source = Image.open(img)
     else:
         return None
     
+    print("processing img...")
+    
     objects = detector(source)
     persons = [
         o for o in objects 
         if o['label'] == 'person' and o['score'] > 0.83]
     bboxes = list(map(lambda p: (p['box']['xmin'], p['box']['ymin'], p['box']['xmax'], p['box']['ymax']), persons))
     n = len(persons)
-    
-    print(f"it's got {n} {'ppl' if n > 1 else 'person'} in the image")
+    result = f"it's got {n} {'ppl' if n > 1 else 'person'} in the image"
 
     draw_bbox(bboxes)
     return source
 
+def set_result(*args):
+    global result
+    if result is None:
+        return f"<h1 id='result' style='display: flex;justify-content: center;'>it counts ppl, upload an image, see how it goes</h1>"
+    else:
+        title = f"<h1 id='result' style='display: flex;justify-content: center;'>{result}</h1>"
+        result = None
+        return title
+