uripper commited on
Commit
4dcb901
1 Parent(s): 682358c

adding app

Browse files
Files changed (1) hide show
  1. app.py +17 -0
app.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from diffusers import DiffusionPipeline
3
+ import torch
4
+
5
+ pipeline = DiffusionPipeline.from_pretrained("uripper/GIANNIS", revision="ONNX")
6
+ pipeline.to("cuda")
7
+ im = pipeline(height=1,width=1,num_channels=1, num_inference_steps=1).images[0]
8
+ im.show()
9
+ # When a user clicks button, pipeline will generate an image
10
+
11
+ # def pipeline_interface():
12
+ # pipeline.to("cuda")
13
+ # return pipeline("height=100,width=100").images[0]
14
+
15
+ # iface = gr.Interface(fn=pipeline_interface, inputs=[], outputs="image")
16
+
17
+ # iface.launch()