Spaces:

hyejoo
/

OnDevice-transformers-demo

Paused

hyejoo commited on Jan 17

Commit

86b1e7a

•

1 Parent(s): 187cff1

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("yeye776/t5-OndeviceAI-HomeIoT")
-model = AutoModelForSeq2SeqLM.from_pretrained("yeye776/t5-OndeviceAI-HomeIoT")
 # Gradio 인터페이스 구성
 def generate_answer(input_text):

 # 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):