Update app.py

app.py

@@ -1,6 +1,4 @@
 import streamlit as st
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-import torch
 
 # -------------------------
 # PAGE CONFIG
@@ -26,17 +24,6 @@ def get_category(bmi):
     else:
         return "Obese"
 
-# -------------------------
-# LOAD MODEL
-# -------------------------
-@st.cache_resource
-def load_model():
-    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
-    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
-    return tokenizer, model
-
-tokenizer, model = load_model()
-
 # -------------------------
 # FORM
 # -------------------------