Spaces:
Sleeping
Sleeping
import streamlit as st | |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer | |
# Load the model and tokenizer | |
model_name = "flax-community/t5-recipe-generation" | |
model = AutoModelForSeq2SeqLM.from_pretrained(model_name) | |
tokenizer = AutoTokenizer.from_pretrained(model_name) | |
# Streamlit app for recipe generation | |
st.title("AI-Powered Recipe Generator") | |
prompt = st.text_input("Enter ingredients or a recipe title:") | |
if st.button("Generate Recipe"): | |
inputs = tokenizer(prompt, return_tensors="pt") | |
outputs = model.generate(inputs["input_ids"], max_length=150, num_return_sequences=1) | |
recipe = tokenizer.decode(outputs[0], skip_special_tokens=True) | |
st.write(recipe) | |