ThomasSimonini HF staff commited on
Commit
7c5b6ea
1 Parent(s): ef87e14

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
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