Spaces:

Belligerent
/

word-sense-disambiguation

Running

Belligerent commited on Mar 21, 2022

Commit

205984a

1 Parent(s): 2616037

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
-pipe = pipeline(tokenizer = AutoTokenizer.from_pretrained("jpelhaw/t5-word-sense-disambiguation"), model = AutoModelForSeq2SeqLM.from_pretrained("jpelhaw/t5-word-sense-disambiguation"))
 gr.Interface.from_pipeline(pipe,
             inputs = "text",

 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+pipe = pipeline(model = AutoModelForSeq2SeqLM.from_pretrained("jpelhaw/t5-word-sense-disambiguation"),tokenizer = AutoTokenizer.from_pretrained("jpelhaw/t5-word-sense-disambiguation"), )
 gr.Interface.from_pipeline(pipe,
             inputs = "text",