Update main.py

main.py

@@ -25,23 +25,15 @@ def get():
     data = query({"inputs": "The movie is good"})
     return data
 
-@app.route('/', methods=['POST'])
-def predict():
-    message = "This is good movies" #request.form['message']
-    # choice of the model
-    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)
-
-
-
-# @app.route('/')
-# def home():
-#     print(1)
-#     return {'key':"Hello HuggingFace! Successfully deployed. "}
-#     # model = load_checkpoint('checkpoint.pth')
-#     # print(2)
-#     # res = sample(model, obj.maxlen, 'ap')
-#     # print(3)
-#     # return {'key':res}
+@app.route("/", methods=["GET", "POST"])
+def index():
+    user_input = None
+    response = None
+
+    if request.method == "POST":
+        user_input = request.form.get("user_input")
+        response = query({"inputs": user_input})
+        
+
+    return render_template("index.html", user_input=user_input, response=response)
+