Spaces:
Sleeping
Sleeping
Aaron Vattay commited on
Commit ·
cfd077f
1
Parent(s): 02fec32
am sad
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ import torch.nn as nn
|
|
| 5 |
from PIL import Image
|
| 6 |
from torchvision import transforms
|
| 7 |
import gradio as gr
|
| 8 |
-
device = torch.device("
|
| 9 |
class UPSC(nn.Module):
|
| 10 |
def __init__(self):
|
| 11 |
super(UPSC,self).__init__()
|
|
|
|
| 5 |
from PIL import Image
|
| 6 |
from torchvision import transforms
|
| 7 |
import gradio as gr
|
| 8 |
+
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 9 |
class UPSC(nn.Module):
|
| 10 |
def __init__(self):
|
| 11 |
super(UPSC,self).__init__()
|