latest commit

app.py

@@ -107,6 +107,14 @@ def ner_demo(text):
     return color_coded_text
 
 PROJECT_INTRO = "This is a HF Spaces hosted Gradio App built by NLP Group 27 . The model has been trained on surrey-nlp/PLOD-CW dataset"
+
+def echo(text, request: gr.Request):
+    if request:
+        print("Request headers dictionary:", request.headers)
+        print("IP address:", request.client.host)
+        print("Query parameters:", dict(request.query_params))
+    return text
+
 # Create the Gradio interface
 demo = gr.Interface(
     fn=ner_demo,
@@ -116,6 +124,22 @@ demo = gr.Interface(
     title="Named Entity Recognition on PLOD-CW ",
     description=f"{PROJECT_INTRO}\n\nEnter text to extract named entities using a NER model."
 )
+'''
+with gr.Blocks() as demo:
+    gr.Markdown("# Page Title")
+    gr.Markdown("## Subtitle with h2 Font")
+    inputs=gr.Textbox(lines=10, placeholder="Enter text here...", label="Input Text")
+
+    with gr.Column():
+        echo_output = gr.Textbox(label="Echo Output")
+        html_output = ner_demo
+
+    with gr.Column():
+        button1 = gr.Button("Submit")
+'''            
+
+CUSTOM_PATH = "/gradio"
+app = gr.mount_gradio_app(app, demo, path=CUSTOM_PATH)
 
 # Function to run FastAPI
 def run_fastapi():
@@ -125,7 +149,7 @@ def run_fastapi():
 def run_gradio():
     demo.launch(server_name="0.0.0.0", server_port=7860)
 
-# Run both servers in separate threads
-threading.Thread(target=run_fastapi).start()
-threading.Thread(target=run_gradio).start()
+
+run_fastapi()
+#threading.Thread(target=run_gradio).start()
 

locustfile.py

@@ -0,0 +1,28 @@
+from locust import HttpUser, task, between
+from datasets import load_dataset
+import random 
+datCW = load_dataset("surrey_nlp/PLOD_CW")
+
+
+def pick_random_payload():
+    global datCW
+    idx = random.randint(0, len(datCW["test"])-1)
+    tokens = datCW["test"][idx]["tokens"]
+    payload = ' '.join(tokens)
+    print(payload)
+    return payload
+
+class ApiUser(HttpUser):
+    wait_time = between(1, 5)
+
+    @task
+    def ner_endpoint(self):
+        payload = pick_random_payload()
+        payload = {
+            "text": payload
+        }
+        self.client.post("/ner", json=payload)
+
+if __name__ == "__main__":
+    import os
+    os.system("locust -f locustfile.py")