Spaces:

ysharma
/

Explore_llamav2_with_TGI

Running on CPU Upgrade

ysharma HF staff commited on Jul 18, 2023

Commit

c1127b8

•

1 Parent(s): 09e4eaf

bug fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -37,7 +37,7 @@ def predict(message, chatbot):
     #response_json_object = json.loads(response.text)
     #return response_json_object[0]['generated_text']
-    response = requests.post(url, headers=headers, data=json.dumps(data), auth=('hf', hf_token), stream=True)
     partial_message = ""
     for line in response.iter_lines():

     #response_json_object = json.loads(response.text)
     #return response_json_object[0]['generated_text']
+    response = requests.post(api_url, headers=headers, data=json.dumps(data), auth=('hf', hf_token), stream=True)
     partial_message = ""
     for line in response.iter_lines():