quick testing fix

AudioToText/condensedmodel.py

@@ -22,7 +22,7 @@ from tensorflow import keras
 from keras import layers
 import librosa
 import speech_recognition as sr
-
+from os.path import exists
 # MODEL LOSS
 def CTCLoss(y_true, y_pred):
     # Compute the training-time loss value
@@ -129,7 +129,8 @@ def loadWeights():
     output = "AudioToTextCKPT.hdf5"
 
     # Download
-    gdown.download(url = ckpt_link, output = output, quiet = False, fuzzy = True)
+    if not exists("AudioToTextCKPT.hdf5"):
+        gdown.download(url = ckpt_link, output = output, quiet = False, fuzzy = True)
 
     # Load CKPT to Model
     model.load_weights(output)

TestTranslation/translation.py

@@ -19,6 +19,7 @@ from tensorflow.keras import layers
 from tensorflow.keras.layers.experimental.preprocessing import TextVectorization
 import os
 import gdown
+from os.path import exists
 
 text_file = keras.utils.get_file(
     fname = "spa-eng.zip",
@@ -253,7 +254,8 @@ transformer.summary()
 
 # load weights using gdown
 print(os.listdir())
-gdown.download_folder("https://drive.google.com/drive/folders/1DwN-MlL6MMh7qVJbwoLrWBSMVBN5zbBi")
+if not exists("./EngToSpanishckpts"):
+    gdown.download_folder("https://drive.google.com/drive/folders/1DwN-MlL6MMh7qVJbwoLrWBSMVBN5zbBi")
 transformer.load_weights("./EngToSpanishckpts/cp.ckpt")
 
 spa_vocab = spa_vectorization.get_vocabulary()

TestTranslationChinese/translation_model.py

@@ -17,11 +17,12 @@ import tensorflow as tf
 from tensorflow import keras
 from tensorflow.keras import layers
 from tensorflow.keras.layers.experimental.preprocessing import TextVectorization
-
+from os.path import exists
 import gdown
 
-url = "https://drive.google.com/uc?id=1FOC2x5HlgcFTMgnGhPjvLWWlEqVTLQno"
-gdown.download(url, quiet=False)
+if not exists("cmn.txt"):
+    url = "https://drive.google.com/uc?id=1FOC2x5HlgcFTMgnGhPjvLWWlEqVTLQno"
+    gdown.download(url, quiet=False)
 
 with open('cmn.txt', encoding="utf-8") as f:
     lines = f.read().split("\n")[:-1]
@@ -224,7 +225,8 @@ class TransformerDecoder(layers.Layer):
         return config
 
 url = "https://drive.google.com/uc?id=1a4eTAL4sLUi42P28Veihrv-fVPwFymTa"
-gdown.download(url, quiet=False)
+if not exists("re-model.h5"):
+    gdown.download(url, quiet=False)
 
 custom_objects = {"TransformerEncoder": TransformerEncoder, "PositionalEmbedding": PositionalEmbedding, "TransformerDecoder": TransformerDecoder}
 with keras.utils.custom_object_scope(custom_objects):

TextToAudio/test.py

@@ -1,5 +0,0 @@
-import subprocess
-
-subprocess.call('ln -sf /opt/bin/nvidia-smi /usr/bin/nvidia-smi')
-
-print(subprocess.getoutput('nvidia-smi'))