kcarnold commited on
Commit
e12a53c
1 Parent(s): 4d5d436
Files changed (1) hide show
  1. app.py +5 -1
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
+ """