Spaces:

Evelyn18
/

space-becas

Runtime error

Evelyn18 commited on Jul 27, 2022

Commit

f8a3f8d

•

1 Parent(s): 63f0411

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,8 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
-tokenizer = AutoTokenizer.from_pretrained("Evelyn18/legalectra-base-spanish-finetuned-squad")
-model = AutoModelForCausalLM.from_pretrained("Evelyn18/legalectra-base-spanish-finetuned-squad")
 def predict(input, history=[]):
     # tokenize the new input sentence

 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+tokenizer = AutoTokenizer.from_pretrained("Evelyn18/roberta-base-spanish-squades-becasv3")
+model = AutoModelForCausalLM.from_pretrained("Evelyn18/roberta-base-spanish-squades-becasv3")
 def predict(input, history=[]):
     # tokenize the new input sentence