Update app.py

app.py

@@ -4,6 +4,33 @@ import pyewts
 
 converter = pyewts.pyewts()
 
+def remove_repeated_words(text):
+    # Tokenize the input text into words
+    words = text.split()
+
+    # Create a dictionary to count word occurrences
+    word_count = {}
+
+    # Create a list to store the final words
+    new_words = []
+
+    for word in words:
+        # Check if the word is in the dictionary
+        if word in word_count:
+            # If it has occurred once before, add it to the list with a count of 2
+            if word_count[word] == 1:
+                new_words.append(word)
+                word_count[word] = 2
+        else:
+            # If it has not occurred before, add it to the dictionary with a count of 1
+            word_count[word] = 1
+            new_words.append(word)
+
+    # Join the modified words back into a string
+    result = ' '.join(new_words)
+    return result
+
+# Example usage
 
 
 pipe = pipeline(model="spsither/whisper-small-r2-70k-2ep",device='cuda')  # change to "your-username/the-name-you-picked"