Update pyai.py

pyai.py

@@ -3,6 +3,7 @@ import whisper
 import numpy as np
 from torch import nn
 from torch import Tensor
+from transformers import pipeline
 from sklearn.tree import DecisionTreeRegressor
 
 
@@ -60,6 +61,15 @@ class PyAI:
         else:
             soft = nn.Softmax(dim=1).to(self.GPU)(x)
             return soft
+        
+    def Sigmoid(self, x: list | Tensor):
+        if isinstance(x, list):
+            tensor = Tensor(x, 1).to(self.GPU)
+            sigmod = nn.Sigmoid().to(self.GPU)(tensor)
+            return sigmod
+        else:
+            sigmod = nn.Sigmoid().to(self.GPU)(x)
+            return sigmod
 
     def decisionTree(self, trainX: list, trainY: list, words: list):
         w = np.array([len(a) for a in words]).reshape(-1, 1)
@@ -149,4 +159,13 @@ class PyAI:
 
         def getPartOfSpeech(self, text: str):
             POS = spacy.load("en_core_web_sm")
-            return POS(text)[0].tag_
+            return POS(text)[0].tag_
+        
+        def BERT(self, text: str, model: str = "bert-base-uncased"):
+            BERT = pipeline("fill-mask", model=model)
+            return BERT(text)
+        
+    class Transformers:
+        def __init__(self, text: str, typeOfOperation: str = "text-generation", model: str = "gpt2"):
+            transformer = pipeline(typeOfOperation, model=model)
+            return transformer(text)