erfan226 commited on
Commit
3489839
1 Parent(s): f16722b

Add PSFT Model

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -1,6 +1,10 @@
1
  import streamlit as st
2
- from transformers import (T5ForConditionalGeneration, AutoTokenizer, pipeline)
3
- import torch
 
 
 
 
4
 
5
  model_path = 'erfan226/persian-t5-formality-transfer'
6
  model = T5ForConditionalGeneration.from_pretrained(model_path)
 
1
  import streamlit as st
2
+ # from transformers import (T5ForConditionalGeneration, AutoTokenizer, pipeline)
3
+ from transformers import pipeline
4
+
5
+ # import torch
6
+
7
+ st.set_page_config(layout="wide", page_title="تبدیل متون محاوره‌ای به فارسی")
8
 
9
  model_path = 'erfan226/persian-t5-formality-transfer'
10
  model = T5ForConditionalGeneration.from_pretrained(model_path)