johann22 commited on
Commit
c72498d
β€’
1 Parent(s): 1db05ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -11,6 +11,8 @@ import urllib.request
11
  import uuid
12
  import requests
13
  import io
 
 
14
  loaded_model=[]
15
  for i,model in enumerate(models):
16
  loaded_model.append(gr.load(f'models/{model}'))
@@ -19,7 +21,8 @@ print (loaded_model)
19
  now = datetime.now()
20
  date_time_str = now.strftime("%Y-%m-%d %H:%M:%S")
21
 
22
- client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
23
  history = []
24
 
25
  def format_prompt(message, history):
 
11
  import uuid
12
  import requests
13
  import io
14
+ import chat_models
15
+
16
  loaded_model=[]
17
  for i,model in enumerate(models):
18
  loaded_model.append(gr.load(f'models/{model}'))
 
21
  now = datetime.now()
22
  date_time_str = now.strftime("%Y-%m-%d %H:%M:%S")
23
 
24
+ client = InferenceClient(chat_models.models[0])
25
+ #client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
26
  history = []
27
 
28
  def format_prompt(message, history):