modified handler

handler.py

@@ -7,72 +7,72 @@ import shutil
 from TTS.utils.synthesizer import Synthesizer
 
 
-def get_or_model():
-    # URL of the ZIP file to download
-    if os.path.exists('./or'):
-        return
-    zip_file_url = 'https://github.com/AI4Bharat/Indic-TTS/releases/download/v1-checkpoints-release/or.zip'
-
-    # Send a GET request to download the file
-    response = requests.get(zip_file_url)
-
-    try:
-        if response.status_code == 200:
-            # Extract the content from the response
-            zip_data = io.BytesIO(response.content)
+# def get_or_model():
+#     # URL of the ZIP file to download
+#     if os.path.exists('./or'):
+#         return
+#     zip_file_url = 'https://github.com/AI4Bharat/Indic-TTS/releases/download/v1-checkpoints-release/or.zip'
+
+#     # Send a GET request to download the file
+#     response = requests.get(zip_file_url)
+
+#     try:
+#         if response.status_code == 200:
+#             # Extract the content from the response
+#             zip_data = io.BytesIO(response.content)
             
-            # Unzip the file
-            with zipfile.ZipFile(zip_data, 'r') as zip_ref:
-                # Replace 'path_to_extract' with the directory path where you want to extract the contents
-                zip_ref.extractall('./')
-            print("ZIP file downloaded and extracted successfully.")
-        else:
-            print("Failed to download ZIP file.")
-    except Exception as e:
-        print(f"Error occurred while downloading and extracting ZIP file: {e}")
-
-def get_indic_tts_library():
-
-    if os.path.exists("./IndicTTS"):
-        print("IndicTTS library already exists, skipping download.")
-        return
-
-    # GitHub repository URL
-    github_repo_url = 'https://github.com/TRY-ER/IndicTTSCustom.git'
-
-    # Prepare target directory
-    try:
-        os.makedirs("./IndicTTS", exist_ok=True)
-    except OSError as e:
-        print(f"Error creating directory: {e}")
-        return
-
-    # Execute the git clone command
-    try:
-        subprocess.run(['git', 'clone', github_repo_url, './IndicTTS'], check=True)
-        print("IndicTTS library downloaded successfully.")
-    except subprocess.CalledProcessError as e:
-        print(f"Failed to download IndicTTS library: {e}")
-
-def replace_json():
-
-    # Source file path
-    source_file = './IndicTTSCustom/custom_config.json'
-
-    # Destination file path (where you want to move the file)
-    destination_file = './or/fastpitch/config.json'
-
-    # Move the file
-    try:
-        # Replace the file if it already exists at the destination
-        shutil.copy(source_file, destination_file)
-        print(f"File moved from {source_file} to {destination_file} successfully.")
-    except FileNotFoundError:
-        print("Source file not found.")
-    except PermissionError:
-        print("Permission denied.")
-    except shutil.Error as e:
-        print(f"Error: {e}")
+#             # Unzip the file
+#             with zipfile.ZipFile(zip_data, 'r') as zip_ref:
+#                 # Replace 'path_to_extract' with the directory path where you want to extract the contents
+#                 zip_ref.extractall('./')
+#             print("ZIP file downloaded and extracted successfully.")
+#         else:
+#             print("Failed to download ZIP file.")
+#     except Exception as e:
+#         print(f"Error occurred while downloading and extracting ZIP file: {e}")
+
+# def get_indic_tts_library():
+
+#     if os.path.exists("./IndicTTS"):
+#         print("IndicTTS library already exists, skipping download.")
+#         return
+
+#     # GitHub repository URL
+#     github_repo_url = 'https://github.com/TRY-ER/IndicTTSCustom.git'
+
+#     # Prepare target directory
+#     try:
+#         os.makedirs("./IndicTTS", exist_ok=True)
+#     except OSError as e:
+#         print(f"Error creating directory: {e}")
+#         return
+
+#     # Execute the git clone command
+#     try:
+#         subprocess.run(['git', 'clone', github_repo_url, './IndicTTS'], check=True)
+#         print("IndicTTS library downloaded successfully.")
+#     except subprocess.CalledProcessError as e:
+#         print(f"Failed to download IndicTTS library: {e}")
+
+# def replace_json():
+
+#     # Source file path
+#     source_file = './IndicTTSCustom/custom_config.json'
+
+#     # Destination file path (where you want to move the file)
+#     destination_file = './or/fastpitch/config.json'
+
+#     # Move the file
+#     try:
+#         # Replace the file if it already exists at the destination
+#         shutil.copy(source_file, destination_file)
+#         print(f"File moved from {source_file} to {destination_file} successfully.")
+#     except FileNotFoundError:
+#         print("Source file not found.")
+#     except PermissionError:
+#         print("Permission denied.")
+#     except shutil.Error as e:
+#         print(f"Error: {e}")
 
 
 # def pred_audio():
@@ -105,16 +105,14 @@ def replace_json():
 #     with open("odia_audio_female.wav", "wb") as f:
 #         f.write(byte_io.read())
 
-
-get_or_model()
-get_indic_tts_library()
-replace_json()
-
 from IndicTTS.inference.src.inference import TextToSpeechEngine
+import torch
 
 class EndpointHandler():
     def __init__(self,path=""):
         self.path = path
+        print(f"python version>> ",{torch.__version__})
+        print(f"cuda version>> ",{torch.cuda.get_device_capability(0)[0]})
         models = {}
         odia_model = Synthesizer(
             tts_checkpoint="./or/fastpitch/best_model.pth",
@@ -134,7 +132,7 @@ class EndpointHandler():
         output_lang_code = data["inputs"]["output_lang_code"]
         speaker_gender = data["inputs"]["speaker_gender"]
         audio_array= self.engine.infer_from_text(
-        input_text=input,
+        input_text=input_text,
         lang=output_lang_code,
         speaker_name=speaker_gender,
         )