| import streamlit as st | |
| from tungstenkit import models | |
| # Load Tungsten AI model | |
| model = models.get("text-to-image") | |
| # Streamlit app title | |
| st.title("Text to Image Generation with Tungsten AI") | |
| # User input for text | |
| text_input = st.text_area("Enter text for image generation:") | |
| # Generate Image button | |
| if st.button("Generate Image"): | |
| # Generate image based on input text | |
| result = model.predict({"prompt": text_input}) | |
| # Get the image path from the result | |
| image_path = result["image"] | |
| # Display the generated image | |
| st.image(image_path, caption="Generated Image", use_column_width=True) |