EmoCube commited on
Commit
29838fe
·
verified ·
1 Parent(s): 90d5300

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -76,12 +76,12 @@ def run_huggingface_model(model, messages, max_tokens, temperature, top_p):
76
 
77
  model = "https://api-inference.huggingface.co/models/" + model + "/v1/chat/completions";
78
  response = requests.post(model, headers=headers, json=payload, timeout=30)
79
- print(response)
80
- response = json.loads(response)
81
 
82
- if hasattr(response, 'error'):
83
  print("ERROR: " + response.error)
84
  else:
 
 
85
  result = response.choices[0].message.content
86
  return result
87
 
 
76
 
77
  model = "https://api-inference.huggingface.co/models/" + model + "/v1/chat/completions";
78
  response = requests.post(model, headers=headers, json=payload, timeout=30)
 
 
79
 
80
+ if response.status_code != 200:
81
  print("ERROR: " + response.error)
82
  else:
83
+ print(response.content)
84
+ response = response.content
85
  result = response.choices[0].message.content
86
  return result
87