Update app.py

app.py

@@ -37,8 +37,22 @@ def write_elo_ratings(elo_ratings):
             json.dump(elo_ratings, file, indent=4)
 
 def get_bot_response(url, prompt):
-    # ... (same as before, but add error handling)
-
+    payload = {
+        "input": {
+            "prompt": prompt,
+            "sampling_params": {
+                "max_new_tokens": 16,
+                "temperature": 0.7,
+            }
+        }
+    }
+    headers = {
+        "accept": "application/json",
+        "content-type": "application/json",
+        "authorization": os.environ.get("RUNPOD_TOKEN")
+    }
+    response = requests.post(url, json=payload, headers=headers)
+    return response.json()
 def chat_with_bots(user_input):
     bot_names = list(chatbots.keys())
     random.shuffle(bot_names)