Update app.py

app.py

@@ -5,21 +5,52 @@ import zipfile
 import requests
 from TTS.api import TTS
 
-#... (rest of the code remains the same)
+# Verifica se è disponibile una GPU, altrimenti utilizza la CPU
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
-# Add a new section for model sharing
-model_sharing_section = gr.components.Section("Model Sharing")
+os.environ["COQUI_TOS_AGREED"] = "1"
 
-model_input = gr.components.Textbox(label="Share your model/link")
-model_output = gr.components.Markdown(label="Shared Models")
+tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to(device)
 
-model_sharing_iface = gr.Interface(
- fn=lambda x: x, # dummy function to display the input
- inputs=model_input,
- outputs=model_output,
- title="Share Your Model",
- description="Share your model or link with the community!"
-)
+def clone(text, url, language):
+ response = requests.get(url)
+
+ with open("temp.zip", "wb") as f:
+ f.write(response.content)
+
+ with zipfile.ZipFile("temp.zip", "r") as zip_ref:
+ zip_ref.extractall()
+
+ audio_file = [f for f in os.listdir(".") if f.endswith(".wav")][0]
+
+ tts.tts_to_file(text=text, speaker_wav=audio_file, language=language, file_path="./output.wav")
+
+ os.remove(audio_file)
+ os.remove("temp.zip")
+
+ return "./output.wav"
+
+languages = [
+ "en", "es", "fr", "de", "it",
+ "pt", "pl", "tr", "ru", "nl",
+ "cs", "ar", "zh-cn", "ja", "hu",
+ "ko", "hi"
+]
+
+# Create a separate tab for model sharing
+def model_sharing_tab():
+ model_input = gr.components.Textbox(label="Share your model/link")
+ model_output = gr.components.Markdown(label="Shared Models")
+
+ model_sharing_iface = gr.Interface(
+ fn=lambda x: x, # dummy function to display the input
+ inputs=model_input,
+ outputs=model_output,
+ title="Share Your Model",
+ description="Share your model or link with the community!"
+ )
+
+ model_sharing_iface.launch()
 
 # Combine the two interfaces
 iface = gr.Interface(
@@ -36,9 +67,12 @@ iface = gr.Interface(
  use this colab with caution <3.
  """,
  theme=gr.themes.Base(primary_hue="teal", secondary_hue="teal", neutral_hue="slate"),
- tabs=["Voice Clone", "Model Sharing"]
+ style=gr.Box.vertical,
+ # Add the model sharing tab
+ children=[
+ iface,
+ gr.Blocks(model_sharing_tab)
+ ]
 )
 
-iface.add_component(model_sharing_iface, "Model Sharing")
-
 iface.launch(share=True)