Update main.py

main.py

@@ -1,5 +1,6 @@
 from flask import Flask, jsonify, render_template, request, make_response
 import transformers
+from huggingface_hub import cached_download
 import torch
 from torch import nn
 import re
@@ -12,7 +13,14 @@ from collections import OrderedDict
 app = Flask(__name__)
 
 # create a python dictionary for your models d = {<key>: <value>, <key>: <value>, ..., <key>: <value>}
-dictOfModels = {"BERT" : transformers.pipeline('sentiment-analysis', model="nlptown/bert-base-multilingual-uncased-sentiment")}
+model_url = "nlptown/bert-base-multilingual-uncased-sentiment"
+model_path = cached_download(model_url)
+tokenizer = AutoTokenizer.from_pretrained(model_path)
+
+# Load model using transformers.pipeline
+model = transformers.pipeline(task="sentiment-analysis", model=model_path, tokenizer=tokenizer)
+
+dictOfModels = {"BERT" : model}
 # create a list of keys to use them in the select part of the html code
 listOfKeys = []
 for key in dictOfModels :