import streamlit as st
import replicate

# Set the replicate API token
import os
os.environ["REPLICATE_API_TOKEN"] = "r8_JSR8xlRoCk6cmq3qEOOThVTn3dAgdPq1bWXdj"

st.title("Stable Diffusion Infinite Zoom")

prompt = st.text_area("Enter your prompt here", height=200)
model_name = "arielreplicate/stable_diffusion_infinite_zoom:a2527c5074fc0cf9fa6015a40d75d080d1ddf7082fabe142f1ccd882c18fce61"
output = replicate.run(model_name, input={"prompt": prompt})

st.write("Output:")
st.text_area("Model Output", value=output, height=200)