Update app.py
Browse files
app.py
CHANGED
@@ -13,8 +13,8 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
13 |
tokenizer = AutoTokenizer.from_pretrained("universitytehran/PersianMind-v1.0")
|
14 |
|
15 |
# Conversation template
|
16 |
-
TEMPLATE = "{context}\nYou:
|
17 |
-
CONTEXT = "You are a professional persian to english translator."
|
18 |
|
19 |
# Streamlit app
|
20 |
st.title("PersianMind translator")
|
|
|
13 |
tokenizer = AutoTokenizer.from_pretrained("universitytehran/PersianMind-v1.0")
|
14 |
|
15 |
# Conversation template
|
16 |
+
TEMPLATE = "{context}\nYou: {prompt}\nPersianMind: "
|
17 |
+
CONTEXT = "You are a professional persian to english translator. translate user input to english"
|
18 |
|
19 |
# Streamlit app
|
20 |
st.title("PersianMind translator")
|