Spaces:
Runtime error
Runtime error
Commit
•
241f0c4
1
Parent(s):
c4ea48a
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,7 +30,7 @@ cfg.MODEL.WEIGHTS = "https://layoutlm.blob.core.windows.net/dit/dit-fts/publayne
|
|
| 30 |
|
| 31 |
# Step 3: set device
|
| 32 |
cfg.MODEL.DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 33 |
-
|
| 34 |
# Step 4: define model
|
| 35 |
predictor = DefaultPredictor(cfg)
|
| 36 |
|
|
|
|
| 30 |
|
| 31 |
# Step 3: set device
|
| 32 |
cfg.MODEL.DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 33 |
+
test = ""
|
| 34 |
# Step 4: define model
|
| 35 |
predictor = DefaultPredictor(cfg)
|
| 36 |
|