DaniRIU commited on
Commit
6fc6cae
1 Parent(s): a5b751f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -2,11 +2,11 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
  import gradio as gr
3
  import time
4
 
5
- #godel = gr.load("huggingface/microsoft/GODEL-v1_1-large-seq2seq")
6
- tokenizer = AutoTokenizer.from_pretrained("GODEL-v1_1-large-seq2seq")
7
  print("Tokenizer loaded")
8
- model = AutoModelForSeq2SeqLM.from_pretrained("GODEL-v1_1-large-seq2seq", pad_token_id=tokenizer.eos_token_id)
9
  print("Loaded")
 
10
 
11
  def generate(dialog):
12
  dialog = [dialog]
 
2
  import gradio as gr
3
  import time
4
 
5
+ tokenizer = AutoTokenizer.from_pretrained("microsoft/GODEL-v1_1-large-seq2seq")
 
6
  print("Tokenizer loaded")
7
+ model = AutoModelForSeq2SeqLM.from_pretrained("microsoft/GODEL-v1_1-large-seq2seq", pad_token_id=tokenizer.eos_token_id)
8
  print("Loaded")
9
+ #godel = gr.load("huggingface/microsoft/GODEL-v1_1-large-seq2seq")
10
 
11
  def generate(dialog):
12
  dialog = [dialog]