Spaces:

hyejoo
/

OnDevice-transformers-demo

Paused

hyejoo commited on Jan 17

Commit

ccd5efb

•

1 Parent(s): 86b1e7a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,8 +3,8 @@ import gradio as gr
 # Load model directly
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-tokenizer = AutoTokenizer.from_pretrained("models/yeye776/OndeviceAI-T5-v1T")
-model = AutoModelForSeq2SeqLM.from_pretrained("models/yeye776/OndeviceAI-T5-v1")
 # Gradio 인터페이스 구성
 def generate_answer(input_text):

 # Load model directly
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+tokenizer = AutoTokenizer.from_pretrained("yeye776/OndeviceAI-T5-v1T")
+model = AutoModelForSeq2SeqLM.from_pretrained("yeye776/OndeviceAI-T5-v1")
 # Gradio 인터페이스 구성
 def generate_answer(input_text):