fully-app commit
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
|
|
2 |
import torch
|
3 |
import torch.nn as nn
|
4 |
import onnx
|
5 |
-
import
|
6 |
from typing import Tuple, Dict
|
7 |
from train import NUM_CLASSES
|
8 |
from torchvision.models import efficientnet_b0, EfficientNet_B0_Weights
|
|
|
2 |
import torch
|
3 |
import torch.nn as nn
|
4 |
import onnx
|
5 |
+
import data, utils
|
6 |
from typing import Tuple, Dict
|
7 |
from train import NUM_CLASSES
|
8 |
from torchvision.models import efficientnet_b0, EfficientNet_B0_Weights
|