| import streamlit as st | |
| from transformers import pipeline | |
| # Load your model from the Hugging Face Hub | |
| model = pipeline("text-generation", model="noobasuna/instruct-pix2pix-model") | |
| st.title("InstructME2ME") | |
| prompt = st.text_input("Enter your prompt:") | |
| if prompt: | |
| output = model(prompt)[0]["generated_text"] | |
| st.write(output) | |