Spaces:
Sleeping
Sleeping
test
Browse files
app.py
CHANGED
@@ -6,6 +6,9 @@ import torch.nn.functional as F
|
|
6 |
import transformers
|
7 |
import pandas as pd
|
8 |
|
|
|
|
|
|
|
9 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
10 |
|
11 |
from transformers import MarianMTModel, MarianTokenizer
|
@@ -65,4 +68,5 @@ with tokenizer.as_target_tokenizer():
|
|
65 |
|
66 |
|
67 |
st.write(probs_table)
|
68 |
-
st.write(model.config.decoder_start_token_id)
|
|
|
|
6 |
import transformers
|
7 |
import pandas as pd
|
8 |
|
9 |
+
st.title("Streamlit + Transformers")
|
10 |
+
|
11 |
+
"""
|
12 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
13 |
|
14 |
from transformers import MarianMTModel, MarianTokenizer
|
|
|
68 |
|
69 |
|
70 |
st.write(probs_table)
|
71 |
+
st.write(model.config.decoder_start_token_id)
|
72 |
+
"""
|