Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,11 +1,10 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+import torch
 
 st.set_page_config(page_title="FitPlan AI", layout="centered")
 
-# ---------------------------------------------------
 # LOAD MODEL (FIXED FOR FLAN-T5)
-# ---------------------------------------------------
 @st.cache_resource
 def load_model():
     tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")