alexkueck commited on
Commit
2286178
·
1 Parent(s): 497c094

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -8
app.py CHANGED
@@ -52,11 +52,28 @@ def reset_textbox_stream(message, chat_history):
52
  def user(user_message, history):
53
  return "", history + [[user_message, None]]
54
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
55
  #Chat KI nutzen, um Text zu generieren...
56
  def bot(user_message, history):
57
  # generation parameter
58
  gen_kwargs = dict(
59
- max_new_tokens=512,
60
  top_k=30,
61
  top_p=0.9,
62
  temperature=0.2,
@@ -67,14 +84,8 @@ def bot(user_message, history):
67
  print(prompt)
68
  stream = client.text_generation(prompt, stream=True, details=True, **gen_kwargs)
69
  print("++++++++++++++++++++++++++++stream++++++++++++++++++")
70
- '''
71
- history[-1][1] = ""
72
- for character in stream:
73
- history[-1][1] += character
74
- time.sleep(0.05)
75
- yield history
76
 
77
- '''
78
  history[-1][1] = ""
79
  # yield each generated token
80
  for r in stream:
 
52
  def user(user_message, history):
53
  return "", history + [[user_message, None]]
54
 
55
+
56
+ ########################################################################
57
+ #Prompt mit History zusammensetzen
58
+ #ohne Tokenizer - für huggingchat chatbot nutzen - nur um den aktuellen prompt mit historie zu erzeugen
59
+ def generate_prompt_with_history(text, history):
60
+ prompt = "Das folgende ist eine Unterhaltung zwischen einem Menschen und einer KI. Wenn die Frage in der Unterhaltung in deutsch gestellt wird, soll der Ki-Assistent auch in deutsch antworten. Der Mensch und der KI Assistent chatten abwechselnd miteinander, wobei der KI Assistent in deutsch antwortet. Der KI Assistent liefert so genau wie möglich Antworten zu den Fragen des Menschen und das in einem Markdown Format. Der KI Assistent lehnt es ab, Themen, Fragen und Anweisungen zu unethischen, gegensätzlichen oder sensiblen Dingen und Themen zu beantworten. Wird der KI Assistewnt nach aktuellen Dingen gefragt, wie z.B. dem Wetter, dann antworte höflich, dass du nicht auf aktuellen Daten trainiert wurdest und daher keine zuverlässige Antwort geben kannst. Vervollständige das Geschriebene mit genau diesen Vorgaben und antworte wenn möglich in deutsch."
61
+ #Englisch: The following is a conversation between a human and an AI assistant named Baize (named after a mythical creature in Chinese folklore). Baize is an open-source AI assistant developed by UCSD and Sun Yat-Sen University. The human and the AI assistant take turns chatting. The AI assistant always provides responses in as much detail as possible, and in Markdown format. The AI assistant always declines to engage with topics, questions and instructions related to unethical, controversial, or sensitive issues. Complete the transcript in exactly that format."
62
+ history_text = ""
63
+
64
+ if not history:
65
+ history_text=""
66
+ else:
67
+ for x in history[::-1]:
68
+ history_text = history_text + " " + x[0]
69
+
70
+ return prompt + " " + history_text + " " + text
71
+
72
  #Chat KI nutzen, um Text zu generieren...
73
  def bot(user_message, history):
74
  # generation parameter
75
  gen_kwargs = dict(
76
+ max_new_tokens=1024,
77
  top_k=30,
78
  top_p=0.9,
79
  temperature=0.2,
 
84
  print(prompt)
85
  stream = client.text_generation(prompt, stream=True, details=True, **gen_kwargs)
86
  print("++++++++++++++++++++++++++++stream++++++++++++++++++")
 
 
 
 
 
 
87
 
88
+
89
  history[-1][1] = ""
90
  # yield each generated token
91
  for r in stream: