Update main.py

main.py

@@ -32,7 +32,7 @@ def get():
 def predict():
     message = "This is good movies" #request.form['message']
     # choice of the model
-    results = get_prediction(message, dictOfModels['RoBERTa') # get_prediction(message, dictOfModels['request.form.get("model_choice")'])
+    results = get_prediction(message, dictOfModels['BERT']) # get_prediction(message, dictOfModels['request.form.get("model_choice")'])
     print(f'User selected model : {request.form.get("model_choice")}')
     my_prediction = f'The feeling of this text is {results[0]["label"]} with probability of {results[0]["score"]*100}%.'
     return render_template('result.html', text = f'{message}', prediction = my_prediction)