jonathanlehner commited on
Commit
8d12f05
1 Parent(s): 15b5b7a
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,9 +1,9 @@
1
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM
2
 
3
- tokenizer = AutoTokenizer.from_pretrained("bigscience/T0pp")
4
- model = AutoModelForSeq2SeqLM.from_pretrained("bigscience/T0pp")
5
- #tokenizer = AutoTokenizer.from_pretrained("ethzanalytics/ai-msgbot-gpt2-M")
6
- #model = AutoModelForCausalLM.from_pretrained("ethzanalytics/ai-msgbot-gpt2-M")
7
 
8
  import gradio as gr
9
  import random
 
1
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM
2
 
3
+ #tokenizer = AutoTokenizer.from_pretrained("bigscience/T0pp")
4
+ #model = AutoModelForSeq2SeqLM.from_pretrained("bigscience/T0pp")
5
+ tokenizer = AutoTokenizer.from_pretrained("ethzanalytics/ai-msgbot-gpt2-M")
6
+ model = AutoModelForCausalLM.from_pretrained("ethzanalytics/ai-msgbot-gpt2-M")
7
 
8
  import gradio as gr
9
  import random