Gizachew commited on
Commit
852aa43
·
verified ·
1 Parent(s): eb2093a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "detection_model.pth" # Ensure this path matches where the model is stored
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