Update app.py

app.py

@@ -7,6 +7,10 @@ app = Flask(__name__)
 # Initialize the Gradio client
 gradio_client = Client("bhlewis/BERT-for-Patents_Semantic-Patent-Finder-v2")
 
+@app.route('/')
+def home():
+    return "Server is running. Use /api/predict for predictions."
+
 @app.route('/api/predict', methods=['POST'])
 def predict():
     try:
@@ -18,8 +22,8 @@ def predict():
         print(f"Top K: {top_k}")
 
         result = gradio_client.predict(
-            query,
-            top_k,
+            query=query,
+            top_k=top_k,
             api_name="/predict"
         )