Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -2,6 +2,11 @@
|
|
2 |
|
3 |
import streamlit as st
|
4 |
import torch
|
|
|
|
|
|
|
|
|
|
|
5 |
|
6 |
def translate(text):
|
7 |
# Tokenize the text
|
|
|
2 |
|
3 |
import streamlit as st
|
4 |
import torch
|
5 |
+
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
6 |
+
|
7 |
+
# Initialize tokenizer and model
|
8 |
+
tokenizer = AutoTokenizer.from_pretrained("your_pretrained_model_name")
|
9 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("your_pretrained_model_name")
|
10 |
|
11 |
def translate(text):
|
12 |
# Tokenize the text
|