Update app.py

app.py

@@ -1,4 +1,17 @@
-import streamlit as st
+from diffusers import StableDiffusionPipeline
+import matplotlib.pyplot as plt
+import torch
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+model_id1 = "dreamlike-art/dreamlike-diffusion-1.0"
+model_id2 = "stabilityai/stable-diffusion-xl-base-1.0"
+
+pipe = StableDiffusionPipeline.from_pretrained(model_id1, torch_dtype=torch.float16, use_safetensors=True)
+pipe = pipe.to("cuda")
+
+prompt = """ batman 
+"""
+image = pipe(prompt).images[0]
+
+print("[PROMPT]: ",prompt)
+plt.imshow(image);
+plt.axis('off');