ysharma HF Staff commited on
Commit
2c77b5e
·
1 Parent(s): 7e230a2
Files changed (1) hide show
  1. app.py +5 -13
app.py CHANGED
@@ -9,26 +9,18 @@ def block_inference(prompt):
9
  generated_text = text_to_text_gm130b(prompt, 1234, 256, 0, 'BeamSearchStrategy', 2, 1, 3, 0.7, 1, 0)
10
  return generated_text
11
 
12
- def chat(message, history):
13
  history = history or []
14
  message = message.lower()
15
  response = block_inference(message)
16
- #if message.startswith("how many"):
17
- # response = random.randint(1, 10)
18
- #elif message.startswith("how"):
19
- # response = random.choice(["Great", "Good", "Okay", "Bad"])
20
- #elif message.startswith("where"):
21
- # response = random.choice(["Here", "There", "Somewhere"])
22
- #else:
23
- # response = "I don't know"
24
  history.append((message, response))
25
  return history, history
26
 
27
- chatbot = gr.Chatbot().style(color_map=("green", "pink"))
28
  demo = gr.Interface(
29
- chat,
30
- ["text", "state"],
31
- [chatbot, "state"],
32
  allow_flagging="never",
33
  )
34
  if __name__ == "__main__":
 
9
  generated_text = text_to_text_gm130b(prompt, 1234, 256, 0, 'BeamSearchStrategy', 2, 1, 3, 0.7, 1, 0)
10
  return generated_text
11
 
12
+ def state_player(message, history):
13
  history = history or []
14
  message = message.lower()
15
  response = block_inference(message)
 
 
 
 
 
 
 
 
16
  history.append((message, response))
17
  return history, history
18
 
19
+ text_editor = gr.Textbox()
20
  demo = gr.Interface(
21
+ state_player,
22
+ [text_editor, "state"],
23
+ [text_editor, "state"],
24
  allow_flagging="never",
25
  )
26
  if __name__ == "__main__":