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 |
|