mou3az commited on
Commit
3d0ab4f
1 Parent(s): 9a5e9c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -158,7 +158,7 @@ def normalize_text(text):
158
  return ' '.join(words)
159
 
160
  llm = HuggingFaceEndpoint(
161
- repo_id="HuggingFaceH4/starchat2-15b-v0.1",
162
  task="text-generation",
163
  max_new_tokens=4096,
164
  temperature=0.6,
@@ -194,9 +194,8 @@ def user3(user_message, history, file_path):
194
  return "", history + [[combined_message, None]], file_path
195
 
196
  messages1 = [
197
- SystemMessage(content="You are a helpful assistant."),
198
- HumanMessage(content="Hi AI, how are you today?"),
199
- AIMessage(content="I'm great thank you. How can I help you?")]
200
  messages2 = messages1.copy()
201
  messages3 = messages1.copy()
202
  messages4 = messages1.copy()
 
158
  return ' '.join(words)
159
 
160
  llm = HuggingFaceEndpoint(
161
+ repo_id="mistralai/Mistral-7B-Instruct-v0.2",
162
  task="text-generation",
163
  max_new_tokens=4096,
164
  temperature=0.6,
 
194
  return "", history + [[combined_message, None]], file_path
195
 
196
  messages1 = [
197
+ HumanMessage(content="You are a helpful assistant."),
198
+ AIMessage(content="Sure, how can I assist you today?")]
 
199
  messages2 = messages1.copy()
200
  messages3 = messages1.copy()
201
  messages4 = messages1.copy()