Karthikraj Sivakumar commited on
Commit
090f43d
·
1 Parent(s): 221b388
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -250,7 +250,7 @@ checkpoint = torch.load('best_model.pth', map_location=device)
250
  model.load_state_dict(checkpoint['model_state_dict'])
251
  model.eval()
252
 
253
- print(f"Model loaded successfully! Using device: {device}")
254
 
255
  # ==========================================
256
  # 4. Prediction Function
@@ -292,7 +292,7 @@ demo = gr.Interface(
292
  fn=predict_captcha,
293
  inputs=gr.Image(type="pil", label="Upload CAPTCHA Image"),
294
  outputs=gr.JSON(label="Prediction Results"),
295
- title="🔐 CAPTCHA Recognition System",
296
  description="""
297
  **CS4243 Mini Project - CAPTCHA Recognition using CRNN + CTC Loss**
298
 
 
250
  model.load_state_dict(checkpoint['model_state_dict'])
251
  model.eval()
252
 
253
+ print(f"Model loaded successfully! Using device: {device}")
254
 
255
  # ==========================================
256
  # 4. Prediction Function
 
292
  fn=predict_captcha,
293
  inputs=gr.Image(type="pil", label="Upload CAPTCHA Image"),
294
  outputs=gr.JSON(label="Prediction Results"),
295
+ title="CAPTCHA Recognition System",
296
  description="""
297
  **CS4243 Mini Project - CAPTCHA Recognition using CRNN + CTC Loss**
298