Spaces:
Runtime error
Runtime error
younesbelkada
commited on
Commit
•
a813ad5
1
Parent(s):
4d6b877
add decorator + update reqs
Browse files- app.py +1 -3
- requirements.txt +3 -1
app.py
CHANGED
|
@@ -1,8 +1,6 @@
|
|
| 1 |
import os
|
| 2 |
-
import sys
|
| 3 |
|
| 4 |
import torch
|
| 5 |
-
import cv2
|
| 6 |
import PIL.Image
|
| 7 |
import numpy as np
|
| 8 |
import gradio as gr
|
|
@@ -49,7 +47,7 @@ def get_generator(model_name):
|
|
| 49 |
generator = generator.cuda()
|
| 50 |
return generator
|
| 51 |
|
| 52 |
-
|
| 53 |
def inference(seed, choice, model_name, coef, nb_images=NB_IMG):
|
| 54 |
np.random.seed(seed)
|
| 55 |
|
|
|
|
| 1 |
import os
|
|
|
|
| 2 |
|
| 3 |
import torch
|
|
|
|
| 4 |
import PIL.Image
|
| 5 |
import numpy as np
|
| 6 |
import gradio as gr
|
|
|
|
| 47 |
generator = generator.cuda()
|
| 48 |
return generator
|
| 49 |
|
| 50 |
+
@torch.no_grad()
|
| 51 |
def inference(seed, choice, model_name, coef, nb_images=NB_IMG):
|
| 52 |
np.random.seed(seed)
|
| 53 |
|
requirements.txt
CHANGED
|
@@ -1,2 +1,4 @@
|
|
| 1 |
torch
|
| 2 |
-
numpy
|
|
|
|
|
|
|
|
|
| 1 |
torch
|
| 2 |
+
numpy
|
| 3 |
+
opencv-python
|
| 4 |
+
Pillow
|