Update app.py

app.py

@@ -1,13 +1,10 @@
 import requests
 import os
 import streamlit as st
-import torch
 from parler_tts import ParlerTTSForConditionalGeneration
 from transformers import AutoTokenizer
 import soundfile as sf
 
-# print(torch.cuda.is_available())
-
 # load_dotenv(find_dotenv())
 # HUGGINGFACEHUB_API_TOKEN = os.getenv("token")
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
@@ -65,7 +62,7 @@ def generate_story(scene):
 
 def gen_audio(message):
 
-    device = "cuda:0" if torch.cuda.is_available() else "cpu"
+    device = "cpu"
 
     model = ParlerTTSForConditionalGeneration.from_pretrained("parler-tts/parler_tts_mini_v0.1").to(device)
     tokenizer = AutoTokenizer.from_pretrained("parler-tts/parler_tts_mini_v0.1")