Update main.py

main.py

@@ -43,7 +43,7 @@ def classify_msg(message):
         response.text[response.text.find("{") : response.text.rfind("}") + 1]
     )
 
-    return gr.Label(json_response['category'])
+    return json_response['category']
 
     
 @app.get("/")