leo-huovi commited on
Commit
4bf70f6
1 Parent(s): ffc8c9c

Update main.py

Browse files

If-else kun tyhjä merkkijono

Files changed (1) hide show
  1. main.py +12 -7
main.py CHANGED
@@ -3,29 +3,34 @@ from flask import Flask, render_template, request
3
  import random
4
  from transformers import pipeline
5
 
6
-
7
-
8
  os.environ['SENTENCE_TRANSFORMERS_HOME'] = './.cache'
9
 
10
-
11
  pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fi")
12
 
13
  app = Flask(__name__)
14
 
 
15
  @app.route('/')
16
  def home():
17
  return render_template('index.html')
18
 
19
-
20
  @app.route('/submit', methods=['POST'])
21
  def submit():
22
  kaannos_pyynto = request.form["translation_request" ]
23
 
24
- random_number = random.randint(0,100)
 
25
 
26
- kaannos = pipe(kaannos_pyynto)
27
 
28
- return kaannos[0]['translation_text']
 
 
 
 
 
29
 
30
  if __name__ == '__main__':
31
  app.run(debug=True)
 
3
  import random
4
  from transformers import pipeline
5
 
6
+ #Cachen kanssa oli ongelmia, tällä lähti toimimaan
 
7
  os.environ['SENTENCE_TRANSFORMERS_HOME'] = './.cache'
8
 
9
+ #Helsinki-NLP-malli
10
  pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fi")
11
 
12
  app = Flask(__name__)
13
 
14
+ #Flask-api GET
15
  @app.route('/')
16
  def home():
17
  return render_template('index.html')
18
 
19
+ #Flast-api POST
20
  @app.route('/submit', methods=['POST'])
21
  def submit():
22
  kaannos_pyynto = request.form["translation_request" ]
23
 
24
+ #Testi alussa
25
+ #random_number = random.randint(0,100)
26
 
 
27
 
28
+ #Malli toisti puutaheinäää kun sille lähetti tyhjän merkkijonon....
29
+ if len(kaannos_pyynto) > 0:
30
+ kaannos = pipe(kaannos_pyynto)
31
+ return kaannos[0]['translation_text']
32
+ else:
33
+ return ""
34
 
35
  if __name__ == '__main__':
36
  app.run(debug=True)