sotosbarl commited on
Commit
728dd2c
·
verified ·
1 Parent(s): 247f734

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -12,7 +12,8 @@ def init_session_state():
12
 
13
  # Initialize session state
14
  init_session_state()
15
- pipe = pipeline("text2text-generation", model="google/flan-t5-base")
 
16
  # model_name = "MoritzLaurer/mDeBERTa-v3-base-mnli-xnli"
17
  # tokenizer = AutoTokenizer.from_pretrained(model_name)
18
  # model = AutoModelForSequenceClassification.from_pretrained(model_name)
@@ -60,7 +61,7 @@ if text:
60
  translator = Translator(from_lang='el', to_lang='en')
61
  translator2 = Translator(from_lang='en', to_lang='el')
62
 
63
- st.text("H ερώτηση σας σχετίζεται με " + output+"δίκαιο")
64
 
65
 
66
  # Translate the text from Greek to English
 
12
 
13
  # Initialize session state
14
  init_session_state()
15
+ # pipe = pipeline("text2text-generation", model="google/flan-t5-base")
16
+ pipe = pipeline("text-generation", model="GeneZC/MiniChat-1.5-3B")
17
  # model_name = "MoritzLaurer/mDeBERTa-v3-base-mnli-xnli"
18
  # tokenizer = AutoTokenizer.from_pretrained(model_name)
19
  # model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
61
  translator = Translator(from_lang='el', to_lang='en')
62
  translator2 = Translator(from_lang='en', to_lang='el')
63
 
64
+ st.text("H ερώτηση σας σχετίζεται με " + output+ " δίκαιο")
65
 
66
 
67
  # Translate the text from Greek to English