Update app.py

app.py

@@ -58,18 +58,19 @@ def api_predict(text):
     words = get_hate_words(text)
     return {"hate_words": words}
 
-# UI setup
-with gr.Blocks() as demo:
+# Create the main Gradio UI
+with gr.Blocks() as gradio_ui:
     input_text = gr.Textbox(label="Enter Georgian text")
     output_json = gr.JSON(label="Detected Hate Words")
     submit_btn = gr.Button("Predict")
-    
     submit_btn.click(fn=api_predict, inputs=input_text, outputs=output_json)
 
-# FastAPI app for exposing /predict
+# Create FastAPI app
 app = FastAPI()
-app = mount_gradio_app(app, demo, path="/")  # UI at /
 
-# Extra endpoint at /predict (external API POST)
-api_iface = gr.Interface(fn=api_predict, inputs=gr.Textbox(), outputs="json")
-app = mount_gradio_app(app, api_iface, path="/predict")  # POST at /predict
+# Mount Gradio UI at `/`
+app = mount_gradio_app(app, gradio_ui, path="/")
+
+# Mount Gradio API interface at `/predict`
+api_interface = gr.Interface(fn=api_predict, inputs=gr.Textbox(), outputs="json")
+app = mount_gradio_app(app, api_interface, path="/predict")