Spaces:

Patinya
/

Patin566

Runtime error

Patinya commited on Oct 11, 2022

Commit

67f6b07

•

1 Parent(s): b5d77b6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import pipeline
 #model = AutoModelForSeq2SeqLM.from_pretrained("t5-base")
 #tokenizer = AutoTokenizer.from_pretrained("t5-base")
-classifier = pipeline("translation", model="t5-base")
 def main():
     st.title("Text translation")

 #model = AutoModelForSeq2SeqLM.from_pretrained("t5-base")
 #tokenizer = AutoTokenizer.from_pretrained("t5-base")
+classifier = pipeline('text-generation', model='KoboldAI/GPT-Neo-2.7B-Shinen')
 def main():
     st.title("Text translation")