Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,11 +27,20 @@ pipe_llama = load_llama_pipeline()
|
|
| 27 |
def get_ingredients_llama(food_name):
|
| 28 |
"""
|
| 29 |
Generate a list of ingredients for the given food item using GPT-Neo.
|
|
|
|
| 30 |
"""
|
| 31 |
-
prompt =
|
|
|
|
|
|
|
|
|
|
| 32 |
try:
|
| 33 |
response = pipe_llama(prompt, max_length=50, num_return_sequences=1)
|
| 34 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 35 |
except Exception as e:
|
| 36 |
return f"Error generating ingredients: {e}"
|
| 37 |
|
|
|
|
| 27 |
def get_ingredients_llama(food_name):
|
| 28 |
"""
|
| 29 |
Generate a list of ingredients for the given food item using GPT-Neo.
|
| 30 |
+
Returns a clean, comma-separated list of ingredients.
|
| 31 |
"""
|
| 32 |
+
prompt = (
|
| 33 |
+
f"List the main ingredients typically used to prepare {food_name}. "
|
| 34 |
+
"Provide the ingredients as a concise, comma-separated list without any explanations."
|
| 35 |
+
)
|
| 36 |
try:
|
| 37 |
response = pipe_llama(prompt, max_length=50, num_return_sequences=1)
|
| 38 |
+
generated_text = response[0]["generated_text"].strip()
|
| 39 |
+
|
| 40 |
+
# Process the response to ensure it's a clean, comma-separated list
|
| 41 |
+
ingredients = generated_text.split(":")[-1].strip() # Handle cases like "Ingredients: ..."
|
| 42 |
+
ingredients = ingredients.replace(".", "").strip() # Remove periods and extra spaces
|
| 43 |
+
return ingredients
|
| 44 |
except Exception as e:
|
| 45 |
return f"Error generating ingredients: {e}"
|
| 46 |
|