man08man commited on
Commit
9919b5d
·
verified ·
1 Parent(s): 3b57444

Delete style.css

Browse files
Files changed (1) hide show
  1. style.css +0 -17
style.css DELETED
@@ -1,17 +0,0 @@
1
- import streamlit as st
2
- from diffusers import StableDiffusionPipeline
3
- import torch
4
-
5
- # Load the model
6
- model_id = "runwayml/stable-diffusion-v1-5"
7
- pipe = StableDiffusionPipeline.from_pretrained(model_id)
8
- pipe = pipe.to("cuda") # Runs on Hugging Face's GPU servers
9
-
10
- # Streamlit UI
11
- st.title("Free AI Image Generator")
12
- prompt = st.text_input("Describe your image...")
13
-
14
- if prompt:
15
- with st.spinner("Generating..."):
16
- image = pipe(prompt).images[0]
17
- st.image(image, caption=prompt, use_column_width=True)