rwitz commited on
Commit
9376219
1 Parent(s): 7ef707c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ def get_bot_response(url, prompt):
47
  "authorization": os.environ.get("RUNPOD_TOKEN")
48
  }
49
  response = requests.post(url, json=payload, headers=headers)
50
- return response.json()
51
 
52
  def chat_with_bots(user_input):
53
  bot_names = list(chatbots.keys())
 
47
  "authorization": os.environ.get("RUNPOD_TOKEN")
48
  }
49
  response = requests.post(url, json=payload, headers=headers)
50
+ return response.json()['output'][0]['generated_text'].replace(prompt,"")
51
 
52
  def chat_with_bots(user_input):
53
  bot_names = list(chatbots.keys())