Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -53,7 +53,7 @@ vectorizer.fit(train_data.text)
|
|
53 |
|
54 |
|
55 |
# Model and transforms preparation
|
56 |
-
model= RNN_model()
|
57 |
# Load state dict
|
58 |
model.load_state_dict(torch.load(
|
59 |
f= 'pretrained_symtom_to_disease_model.pth',
|
|
|
53 |
|
54 |
|
55 |
# Model and transforms preparation
|
56 |
+
model = RNN_model()
|
57 |
# Load state dict
|
58 |
model.load_state_dict(torch.load(
|
59 |
f= 'pretrained_symtom_to_disease_model.pth',
|