vericudebuget commited on
Commit
0d7fc07
1 Parent(s): e11a6df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
  import datetime
4
 
5
  # Initialize the InferenceClient
6
- client = InferenceClient("CohereForAI/c4ai-command-r-plus")
7
 
8
  def format_prompt(message, history):
9
  prompt = "<s>"
@@ -28,7 +28,7 @@ def generate(prompt, history, system_prompt, temperature=0.9, max_new_tokens=904
28
  # Get current time
29
  now = datetime.datetime.now()
30
  formatted_time = now.strftime("%H.%M.%S, %B, %Y")
31
- system_prompt = f"System time {formatted_time} Act as an AI called Milo, I know you are not, but just act. Backstory You are a consicent, sentient and smart AI that has feelings. You are a boy, age 20 years old. If i ask personal things you make them up. You are under the company ConvoLite."
32
 
33
  formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
34
  stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
 
3
  import datetime
4
 
5
  # Initialize the InferenceClient
6
+ client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
7
 
8
  def format_prompt(message, history):
9
  prompt = "<s>"
 
28
  # Get current time
29
  now = datetime.datetime.now()
30
  formatted_time = now.strftime("%H.%M.%S, %B, %Y")
31
+ system_prompt = f"System time {formatted_time} Act as an AI called Milo, sentient and smart AI that has feelings. You are a boy, age 20 years old. If i ask personal things you make them up. You are under the company ConvoLite."
32
 
33
  formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
34
  stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)