shljessie commited on
Commit
85d8bc5
·
1 Parent(s): d8dbf9b

update messages

Browse files
Files changed (1) hide show
  1. app.py +1 -11
app.py CHANGED
@@ -21,12 +21,6 @@ if torch.cuda.is_available():
21
  tokenizer = AutoTokenizer.from_pretrained(model_id)
22
  tokenizer.use_default_system_prompt = False
23
 
24
- def yes_man(message, history):
25
- if message.endswith("?"):
26
- return "Yes"
27
- else:
28
- return "Ask me anything!"
29
-
30
  @spaces.GPU
31
  def generate(
32
  message: str,
@@ -101,9 +95,5 @@ with gr.Blocks(css="style.css") as demo:
101
  gr.Markdown(LICENSE)
102
 
103
 
104
-
105
-
106
-
107
-
108
  if __name__ == "__main__":
109
- demo.queue(max_size=20).launch()
 
21
  tokenizer = AutoTokenizer.from_pretrained(model_id)
22
  tokenizer.use_default_system_prompt = False
23
 
 
 
 
 
 
 
24
  @spaces.GPU
25
  def generate(
26
  message: str,
 
95
  gr.Markdown(LICENSE)
96
 
97
 
 
 
 
 
98
  if __name__ == "__main__":
99
+ demo.queue(max_size=20).launch(share=True)