Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -61,19 +61,6 @@ Instructions:
|
|
| 61 |
"""
|
| 62 |
|
| 63 |
return prompt, bmi, bmi_status
|
| 64 |
-
|
| 65 |
-
# -------------------------
|
| 66 |
-
# Load Model (Small for HF)
|
| 67 |
-
# -------------------------
|
| 68 |
-
|
| 69 |
-
@st.cache_resource
|
| 70 |
-
def load_model():
|
| 71 |
-
tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-small")
|
| 72 |
-
model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-small")
|
| 73 |
-
return tokenizer, model
|
| 74 |
-
|
| 75 |
-
tokenizer, model = load_model()
|
| 76 |
-
|
| 77 |
# -------------------------
|
| 78 |
# User Inputs
|
| 79 |
# -------------------------
|
|
|
|
| 61 |
"""
|
| 62 |
|
| 63 |
return prompt, bmi, bmi_status
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 64 |
# -------------------------
|
| 65 |
# User Inputs
|
| 66 |
# -------------------------
|