Spaces:
Runtime error
Runtime error
jonathanlehner
commited on
Commit
•
8d12f05
1
Parent(s):
15b5b7a
latest
Browse files
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 |
-
|
6 |
-
|
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
|