fffiloni commited on
Commit
563389a
·
1 Parent(s): fe7b043

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import torch
2
  from torch import autocast
3
  from diffusers import StableDiffusionPipeline
@@ -7,5 +8,7 @@ device = "cpu"
7
  pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)
8
  pipe.to(device)
9
 
10
-
11
- print("hello")
 
 
 
1
+ import gradio as gr
2
  import torch
3
  from torch import autocast
4
  from diffusers import StableDiffusionPipeline
 
8
  pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)
9
  pipe.to(device)
10
 
11
+ def process :
12
+ return "hello"
13
+
14
+ gr.Interface(fn=process, inputs="text", outputs="text").launch()