Karthikraj Sivakumar
commited on
Commit
·
090f43d
1
Parent(s):
221b388
fix typos
Browse files
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"
|
| 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="
|
| 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 |
|