Update pyai.py

MitchellShibilski-Unkel · Mar 29, 2024 · c3a67a0 · c3a67a0
1 parent 4305173
commit c3a67a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyai.py b/pyai.py
@@ -32,7 +32,7 @@ def __init__(self, audio: str):
         self.model = whisper.load_model("base")
         self.audio = audio
 
-    def generateTextFromAudio(self):
+    def generateTextFromAudio(self) -> str:
         aud = whisper.load_audio(self.audio)
         aud = whisper.pad_or_trim(aud)
 
@@ -45,7 +45,7 @@ def generateTextFromAudio(self):
 
         return result.text
 
-    def translateText(self, text: str, dataSet: str):
+    def translateText(self, text: str, dataSet: str) -> str:
         with open(dataSet, "r") as d:
             data = d.read()