Update app.py

app.py

@@ -30,6 +30,20 @@ def cut_audio(input_path, output_path, max_duration=30000):
 
     return output_path
 
+def get_text_after_colon(input_text):
+    # Find the first occurrence of ":"
+    colon_index = input_text.find(":")
+    
+    # Check if ":" exists in the input_text
+    if colon_index != -1:
+        # Extract the text after the colon
+        result_text = input_text[colon_index + 1:].strip()
+        return result_text
+    else:
+        # Return the original text if ":" is not found
+        return input_text
+
+
 def solo_xd(prompt):
     images = pipe(prompt=prompt).images[0]
     return images
@@ -75,11 +89,10 @@ def infer(audio_file):
     result = client.predict(
     				llama_q,	# str in 'Message' Textbox component
     				api_name="/predict"
-    )
-
-    
-    
+    )    
     
+    result = get_text_after_colon(result)
+
     print(f"Llama2 result: {result}")
 
     images = pipe(prompt=result).images[0]