Spaces:
Sleeping
Sleeping
try a bit more
Browse files
app.py
CHANGED
|
@@ -8,7 +8,6 @@ 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
|
|
@@ -24,7 +23,7 @@ def get_model(model_name):
|
|
| 24 |
|
| 25 |
tokenizer = get_tokenizer(model_name)
|
| 26 |
model = get_model(model_name)
|
| 27 |
-
|
| 28 |
print(f"The model has {model.num_parameters():,d} parameters.")
|
| 29 |
|
| 30 |
input_text = st.text_input("Enter text to translate", "Hola, mi nombre es Juan")
|
|
|
|
| 8 |
|
| 9 |
st.title("Streamlit + Transformers")
|
| 10 |
|
|
|
|
| 11 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 12 |
|
| 13 |
from transformers import MarianMTModel, MarianTokenizer
|
|
|
|
| 23 |
|
| 24 |
tokenizer = get_tokenizer(model_name)
|
| 25 |
model = get_model(model_name)
|
| 26 |
+
"""
|
| 27 |
print(f"The model has {model.num_parameters():,d} parameters.")
|
| 28 |
|
| 29 |
input_text = st.text_input("Enter text to translate", "Hola, mi nombre es Juan")
|