Updated app.py and requirements

app.py

@@ -10,15 +10,25 @@ from models.bertimbau import BertimbauModel
 # ======================================================
 
 bow_model = BowModel()
-bert_model = BertimbauModel()
 
+bert_model = None
+
+def get_bert_model():
+
+    global bert_model
+
+    if bert_model is None:
+        bert_model = BertimbauModel()
+
+    return bert_model
 
 # ======================================================
 # Inferência
 # ======================================================
-
+@spaces.GPU
 def predict(text):
 
+    global bert_model
     text = text.strip()
 
     if len(text) == 0:
@@ -68,6 +78,7 @@ def predict(text):
     # BERT
     # -------------------------
 
+    bert_model = get_bert_model()
     bert = bert_model.predict(text)
 
     bert_prediction = (

requirements.txt

@@ -1,9 +1,9 @@
 spaces
+gradio>=5.0
+transformers>=4.40
+torch
 datasets
 joblib
 numpy
 pandas
-scikit-learn
-torch
-transformers
-gradio
+scikit-learn