Update app.py

app.py

@@ -2,11 +2,21 @@ import streamlit as st
 import requests
 from transformers import pipeline
 
-# Initialize the AI model for answering plant-related questions
-qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad", framework="pt")
-text_generation_pipeline = pipeline("text-generation", model="gpt2", framework="pt")
+# Cache the AI models to avoid reloading them on every interaction
+@st.cache_resource
+def load_qa_model():
+    return pipeline("question-answering", model="distilbert-base-cased-distilled-squad", framework="pt")
 
-# Function to fetch plant data using OpenFarm API
+@st.cache_resource
+def load_text_generation_model():
+    return pipeline("text-generation", model="gpt2", framework="pt")
+
+# Initialize the AI models
+qa_pipeline = load_qa_model()
+text_generation_pipeline = load_text_generation_model()
+
+# Cache the plant data fetching function to avoid redundant API calls
+@st.cache_data(ttl=3600)  # Cache for 1 hour
 def fetch_plant_data(plant_name):
     url = f"https://openfarm.cc/api/v1/crops?filter={plant_name}"
     response = requests.get(url)