add
Browse files- App_main.py +4 -4
App_main.py
CHANGED
@@ -1,11 +1,11 @@
|
|
1 |
import numpy as np
|
2 |
-
|
3 |
-
import models
|
4 |
-
import gradio as gr
|
5 |
-
|
6 |
os.system('nvidia-smi')
|
7 |
os.system('ls /usr/local')
|
8 |
os.system('pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu116')
|
|
|
|
|
|
|
9 |
import torch
|
10 |
from torchvision import transforms
|
11 |
from torchvision.transforms import InterpolationMode
|
|
|
1 |
import numpy as np
|
2 |
+
import os
|
|
|
|
|
|
|
3 |
os.system('nvidia-smi')
|
4 |
os.system('ls /usr/local')
|
5 |
os.system('pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu116')
|
6 |
+
import models
|
7 |
+
import gradio as gr
|
8 |
+
|
9 |
import torch
|
10 |
from torchvision import transforms
|
11 |
from torchvision.transforms import InterpolationMode
|