Update app.py
Browse files
app.py
CHANGED
|
@@ -19,7 +19,6 @@ if os.path.exists(model_path):
|
|
| 19 |
print(f"Loading saved model {model_path}")
|
| 20 |
model.load_state_dict(torch.load(model_path))
|
| 21 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 22 |
-
)
|
| 23 |
def predict(file_path):
|
| 24 |
max_length = 100000
|
| 25 |
|
|
|
|
| 19 |
print(f"Loading saved model {model_path}")
|
| 20 |
model.load_state_dict(torch.load(model_path))
|
| 21 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
|
| 22 |
def predict(file_path):
|
| 23 |
max_length = 100000
|
| 24 |
|