Spaces:
Runtime error
Runtime error
Update model.py
Browse files
model.py
CHANGED
@@ -13,8 +13,8 @@ from torchvision import transforms
|
|
13 |
from pytorch_grad_cam import GradCAM
|
14 |
|
15 |
# Importing necessary components for the Gradio app
|
16 |
-
from
|
17 |
-
from
|
18 |
|
19 |
|
20 |
def load_model(model_url, model_path):
|
|
|
13 |
from pytorch_grad_cam import GradCAM
|
14 |
|
15 |
# Importing necessary components for the Gradio app
|
16 |
+
from config import config_data
|
17 |
+
from model_architectures import ResNet50, LSTMPyTorch
|
18 |
|
19 |
|
20 |
def load_model(model_url, model_path):
|