Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -1,5 +1,3 @@
|
|
1 |
-
import os
|
2 |
-
|
3 |
import gradio as gr
|
4 |
import pandas as pd
|
5 |
import torch
|
@@ -15,7 +13,6 @@ def predict(img):
|
|
15 |
return " ".join(output[1:-1])
|
16 |
|
17 |
if __name__ == '__main__':
|
18 |
-
os.system('python - m spacy download en_core_web_sm')
|
19 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
20 |
model = CNNtoRNN(
|
21 |
embed_size=256,
|
|
|
|
|
|
|
1 |
import gradio as gr
|
2 |
import pandas as pd
|
3 |
import torch
|
|
|
13 |
return " ".join(output[1:-1])
|
14 |
|
15 |
if __name__ == '__main__':
|
|
|
16 |
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
17 |
model = CNNtoRNN(
|
18 |
embed_size=256,
|