Spaces:

DaniRIU
/

chat

Runtime error

DaniRIU commited on May 18, 2023

Commit

aa15060

1 Parent(s): 9e9e8af

Update model downloading

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,8 +2,9 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import gradio as gr
 import time
-tokenizer = AutoTokenizer.from_pretrained("microsoft/GODEL-v1_1-large-seq2seq")
-model = AutoModelForSeq2SeqLM.from_pretrained("microsoft/GODEL-v1_1-large-seq2seq")
 def generate(dialog):
   dialog = [dialog]

 import gradio as gr
 import time
+godel = gr.Interface.load("microsoft/GODEL-v1_1-large-seq2seq")
+tokenizer = AutoTokenizer.from_pretrained(godel)
+model = AutoModelForSeq2SeqLM.from_pretrained(godel)
 def generate(dialog):
   dialog = [dialog]