Spaces:
Sleeping
Sleeping
Commit ·
7c5b6ea
1
Parent(s): ef87e14
Update app.py
Browse files
app.py
CHANGED
|
@@ -57,6 +57,14 @@ def chat(message,
|
|
| 57 |
}
|
| 58 |
|
| 59 |
print("MODEL" + model)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 60 |
response = requests.post(model, headers=headers, data=json.dumps(data), auth=("hf", hf_token), stream=True)
|
| 61 |
|
| 62 |
|
|
|
|
| 57 |
}
|
| 58 |
|
| 59 |
print("MODEL" + model)
|
| 60 |
+
|
| 61 |
+
if model == "zephyr_7b":
|
| 62 |
+
model = zephyr_7b
|
| 63 |
+
elif model == "llama_7b"
|
| 64 |
+
model = llama_7b
|
| 65 |
+
elif model == "llama_13b":
|
| 66 |
+
model == llama_13b
|
| 67 |
+
|
| 68 |
response = requests.post(model, headers=headers, data=json.dumps(data), auth=("hf", hf_token), stream=True)
|
| 69 |
|
| 70 |
|