tastypear commited on
Commit
add90f0
·
verified ·
1 Parent(s): 31ff5cd

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -2
main.py CHANGED
@@ -99,11 +99,12 @@ def chat_completions():
99
  "data": [prompt, chat_history, system],
100
  "fn_index": fn_index,
101
  "session_hash": session_hash,
 
102
  }
103
 
104
  def generate():
105
- response = requests.post(f"{base_url}/queue/join", json=json_prompt)
106
- url = f"{base_url}/queue/data?session_hash={session_hash}"
107
  data = requests.get(url, stream=True)
108
 
109
  time_now = int(time.time())
 
99
  "data": [prompt, chat_history, system],
100
  "fn_index": fn_index,
101
  "session_hash": session_hash,
102
+ "trigger_id": 11,
103
  }
104
 
105
  def generate():
106
+ response = requests.post(f"{base_url}/gradio_api/queue/join", json=json_prompt)
107
+ url = f"{base_url}/gradio_api/queue/data?session_hash={session_hash}"
108
  data = requests.get(url, stream=True)
109
 
110
  time_now = int(time.time())