Update app.py

app.py

@@ -1,11 +1,8 @@
 import streamlit as st
-from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
+from transformers import pipeline
 from PIL import Image
 import os
 
-# Hugging Face token login (add this as a secret in Hugging Face Spaces)
-os.environ["HF_TOKEN"] = st.secrets["HF_AUTH_TOKEN"]
-
 # Load the image classification pipeline
 @st.cache_resource
 def load_image_classification_pipeline():
@@ -16,22 +13,20 @@ def load_image_classification_pipeline():
 
 pipe_classification = load_image_classification_pipeline()
 
-# Load the Llama model for ingredient generation
+# Load the GPT-Neo model for ingredient generation
 @st.cache_resource
 def load_llama_pipeline():
     """
-    Load the Llama model for ingredient generation.
+    Load the GPT-Neo model for ingredient generation.
     """
-    tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-3B-Instruct", use_auth_token=os.environ["HF_TOKEN"])
-    model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.2-3B-Instruct", use_auth_token=os.environ["HF_TOKEN"])
-    return pipeline("text-generation", model=model, tokenizer=tokenizer)
+    return pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B")
 
 pipe_llama = load_llama_pipeline()
 
-# Function to generate ingredients using the Llama model
+# Function to generate ingredients using GPT-Neo
 def get_ingredients_llama(food_name):
     """
-    Generate a list of ingredients for the given food item using the Llama model.
+    Generate a list of ingredients for the given food item using GPT-Neo.
     """
     prompt = f"List the main ingredients typically used to prepare {food_name}."
     try:
@@ -49,7 +44,7 @@ st.image("IR_IMAGE.png", caption="Food Recognition Model", use_column_width=True
 # Sidebar for model information
 st.sidebar.title("Model Information")
 st.sidebar.write("**Image Classification Model**: Shresthadev403/food-image-classification")
-st.sidebar.write("**LLM for Ingredients**: meta-llama/Llama-3.2-3B-Instruct")
+st.sidebar.write("**LLM for Ingredients**: EleutherAI/gpt-neo-1.3B")
 
 # Upload image
 uploaded_file = st.file_uploader("Choose a food image...", type=["jpg", "png", "jpeg"])