apolinario commited on
Commit
3ce85b9
·
1 Parent(s): 9fbfa70
Files changed (2) hide show
  1. app.py +11 -0
  2. requirements.txt +15 -0
app.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import torch
3
+
4
+ is_cuda = torch.cuda.is_available()
5
+ def greet(name):
6
+ if is_cuda:
7
+ return "Hello cuda" + name + "!!"
8
+ else:
9
+ return "Hello ooops" + name + "!!"
10
+ iface = gr.Interface(fn=greet, inputs="text", outputs="text")
11
+ iface.launch()
requirements.txt ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ git+https://github.com/CompVis/taming-transformers
2
+ ftfy
3
+ regex
4
+ tqdm
5
+ omegaconf
6
+ pytorch-lightning
7
+ torch-fidelity
8
+ transformers
9
+ einops
10
+ gradio
11
+ torch
12
+ torchvision
13
+ Pillow
14
+ numpy
15
+ imageio