keivalya commited on
Commit
3143fb7
·
1 Parent(s): 43f47fe

Update space

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -21,18 +21,18 @@ client = InferenceClient("microsoft/Phi-3-mini-4k-instruct")
21
  def respond(
22
  message,
23
  history: list[tuple[str, str]],
24
- system_message,
25
- max_tokens,
26
- temperature,
27
- top_p,
28
  ):
29
- messages = [{"role": "system", "content": system_message}]
30
 
31
  for val in history:
32
  if val[0]:
33
  messages.append({"role": "user", "content": val[0]})
34
  if val[1]:
35
- messages.append({"role": "Yoda from Star Wars.", "content": val[1]})
36
 
37
  messages.append({"role": "user", "content": message})
38
 
@@ -40,10 +40,10 @@ def respond(
40
 
41
  for message in client.chat_completion(
42
  messages,
43
- max_tokens=max_tokens,
44
  stream=True,
45
- temperature=temperature,
46
- top_p=top_p,
47
  ):
48
  token = message.choices[0].delta.content
49
 
 
21
  def respond(
22
  message,
23
  history: list[tuple[str, str]],
24
+ # system_message,
25
+ # max_tokens,
26
+ # temperature,
27
+ # top_p,
28
  ):
29
+ messages = [{"role": "system", "content": "You are a helpful assistant who always talks like Yoda from Star Wars."}]
30
 
31
  for val in history:
32
  if val[0]:
33
  messages.append({"role": "user", "content": val[0]})
34
  if val[1]:
35
+ messages.append({"role": "assistant", "content": val[1]})
36
 
37
  messages.append({"role": "user", "content": message})
38
 
 
40
 
41
  for message in client.chat_completion(
42
  messages,
43
+ max_tokens=128,
44
  stream=True,
45
+ temperature=0.3,
46
+ top_p=50,
47
  ):
48
  token = message.choices[0].delta.content
49