Update app.py

app.py

@@ -23,7 +23,6 @@ def load_bloom_pipeline():
 
 pipe_bloom = load_bloom_pipeline()
 
-# Function to generate ingredients using BLOOM
 def get_ingredients_bloom(food_name):
     """
     Generate a list of ingredients for the given food item using BLOOM.
@@ -35,7 +34,8 @@ def get_ingredients_bloom(food_name):
         "For example, if the food is pizza, respond with 'cheese, tomato sauce, bread, olive oil, basil'."
     )
     try:
-        response = pipe_bloom(prompt, max_length=50, num_return_sequences=1)
+        # Generate response from the model
+        response = pipe_bloom(prompt, max_new_tokens=50, num_return_sequences=1)
         generated_text = response[0]["generated_text"].strip()
         
         # Post-process the response
@@ -48,8 +48,8 @@ def get_ingredients_bloom(food_name):
 
         return ingredients
     except Exception as e:
+        # Handle any errors that occur during the process
         return f"Error generating ingredients: {e}"
-
 # Streamlit app setup
 st.title("Food Image Recognition with Ingredients")