Akjava commited on
Commit
ad0fa67
Β·
1 Parent(s): 8abaccc
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -46,15 +46,17 @@ def generate_text(messages):
46
  return "Unexpected output format."
47
 
48
 
 
49
  def call_generate_text(message, history):
 
50
  print(message)
51
  print(history)
52
 
53
  #messages = history + message
54
- messages = [{"role":"user","content":message}]
55
  try:
56
- text = generate_text(messages)
57
-
58
  return text
59
  except RuntimeError as e:
60
  print(f"An unexpected error occurred: {e}")
 
46
  return "Unexpected output format."
47
 
48
 
49
+ from gradio import ChatMessage
50
  def call_generate_text(message, history):
51
+ history.append(ChatMessage(role="user", content=message))
52
  print(message)
53
  print(history)
54
 
55
  #messages = history + message
56
+ # messages = [{"role":"user","content":message}]
57
  try:
58
+ text = generate_text(history)
59
+ history.append(ChatMessage(role="assistant", content=text))
60
  return text
61
  except RuntimeError as e:
62
  print(f"An unexpected error occurred: {e}")