From c3a67a04635951f9d48b3519bd243f7b2692cffe Mon Sep 17 00:00:00 2001
From: MitchellShibilski-Unkel
 <133730558+MitchellShibilski-Unkel@users.noreply.github.com>
Date: Fri, 29 Mar 2024 17:51:17 -0600
Subject: [PATCH] Update pyai.py

---
 pyai.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyai.py b/pyai.py
index 8dca809..4bcbef1 100644
--- 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()