EmoCube commited on
Commit
0af01d1
·
verified ·
1 Parent(s): fabfa16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -116,7 +116,7 @@ def query(prompt, is_realistic, num_inference_steps, width, height, mod = None):
116
  if (mod == mod_name):
117
  model = mod_link
118
 
119
- model = "https://api-inference.huggingface.co/models/" + model;
120
 
121
  response = requests.post(model, headers=headers, json=payload, timeout=timeout)
122
  if response.status_code != 200:
 
116
  if (mod == mod_name):
117
  model = mod_link
118
 
119
+ model = f'https://api-inference.huggingface.co/models/"{model.strip}';
120
 
121
  response = requests.post(model, headers=headers, json=payload, timeout=timeout)
122
  if response.status_code != 200: