Update app.py
Browse files
app.py
CHANGED
|
@@ -38,7 +38,7 @@ def generate_code(
|
|
| 38 |
):
|
| 39 |
# call space using its API endpoint
|
| 40 |
url = (
|
| 41 |
-
f"https://hf.space/embed/
|
| 42 |
)
|
| 43 |
r = requests.post(
|
| 44 |
url=url, json={"data": [gen_prompt, max_new_tokens, temperature, seed]}
|
|
|
|
| 38 |
):
|
| 39 |
# call space using its API endpoint
|
| 40 |
url = (
|
| 41 |
+
f"https://hf.space/embed/codeparrot/{model_name.lower()}-subspace/+/api/predict/"
|
| 42 |
)
|
| 43 |
r = requests.post(
|
| 44 |
url=url, json={"data": [gen_prompt, max_new_tokens, temperature, seed]}
|