Update app.py
Browse files
app.py
CHANGED
@@ -163,7 +163,7 @@ def detect_text(image, model, device, max_boxes=100, confidence_threshold=0.5):
|
|
163 |
|
164 |
# Initialize the model
|
165 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
166 |
-
model_path = "
|
167 |
model = load_model(model_path, device)
|
168 |
print("Model loaded successfully.")
|
169 |
|
|
|
163 |
|
164 |
# Initialize the model
|
165 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
166 |
+
model_path = "finetuned_recog_model.pth" # Ensure this path matches where the model is stored
|
167 |
model = load_model(model_path, device)
|
168 |
print("Model loaded successfully.")
|
169 |
|