Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ model = None
|
|
11 |
def initialize_model():
|
12 |
global model
|
13 |
if model is None:
|
14 |
-
model_path = '
|
15 |
if not os.path.exists(model_path):
|
16 |
raise FileNotFoundError(f"Model file not found at {model_path}")
|
17 |
|
|
|
11 |
def initialize_model():
|
12 |
global model
|
13 |
if model is None:
|
14 |
+
model_path = 'model/model_state_dict.pth'
|
15 |
if not os.path.exists(model_path):
|
16 |
raise FileNotFoundError(f"Model file not found at {model_path}")
|
17 |
|