Update app.py
Browse files
app.py
CHANGED
|
@@ -6,15 +6,10 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
|
| 6 |
st.title("Generating Response with HuggingFace Models")
|
| 7 |
st.markdown("## Model: `facebook/blenderbot-400M-distill`")
|
| 8 |
|
| 9 |
-
|
| 10 |
-
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
|
| 14 |
-
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 15 |
-
|
| 16 |
-
os.environ["MODEL"] = "1"#model
|
| 17 |
-
os.environ["TOKENIZER"] = "1"#tokenizer
|
| 18 |
|
| 19 |
|
| 20 |
|
|
|
|
| 6 |
st.title("Generating Response with HuggingFace Models")
|
| 7 |
st.markdown("## Model: `facebook/blenderbot-400M-distill`")
|
| 8 |
|
| 9 |
+
with st.spinner("Getting this ready for you.."):
|
| 10 |
+
model_name = "facebook/blenderbot-400M-distill"
|
| 11 |
+
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
|
| 12 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
|
| 14 |
|
| 15 |
|