Jaward commited on
Commit
a646d0c
·
verified ·
1 Parent(s): 16b934e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -12
app.py CHANGED
@@ -34,22 +34,16 @@ logging.basicConfig(
34
  logger = logging.getLogger(__name__)
35
 
36
  # Set up environment
37
- if os.path.exists("/tmp"):
38
- OUTPUT_DIR = "/tmp/outputs" # Use /tmp for Huggingface Spaces
39
- else:
40
- OUTPUT_DIR = os.path.join(os.getcwd(), "outputs") # Fallback for local dev
41
  os.makedirs(OUTPUT_DIR, exist_ok=True)
42
  logger.info(f"Using output directory: {OUTPUT_DIR}")
43
  os.environ["COQUI_TOS_AGREED"] = "1"
44
 
45
  # Initialize TTS model at the top
46
- try:
47
- device = "cuda" if torch.cuda.is_available() else "cpu"
48
- tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to(device)
49
- logger.info("TTS model initialized on %s", device)
50
- except Exception as e:
51
- logger.error("Failed to initialize TTS model: %s", str(e))
52
- tts = None
53
 
54
  # Define Pydantic model for slide data
55
  class Slide(BaseModel):
@@ -1148,4 +1142,4 @@ with gr.Blocks(title="Agent Feynman") as demo:
1148
  )
1149
 
1150
  if __name__ == "__main__":
1151
- demo.launch(allowed_paths=[OUTPUT_DIR], max_file_size="5mb")
 
34
  logger = logging.getLogger(__name__)
35
 
36
  # Set up environment
37
+ OUTPUT_DIR = os.path.join(os.getcwd(), "outputs") # Fallback for local dev
 
 
 
38
  os.makedirs(OUTPUT_DIR, exist_ok=True)
39
  logger.info(f"Using output directory: {OUTPUT_DIR}")
40
  os.environ["COQUI_TOS_AGREED"] = "1"
41
 
42
  # Initialize TTS model at the top
43
+
44
+ device = "cuda" if torch.cuda.is_available() else "cpu"
45
+ tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to(device)
46
+ logger.info("TTS model initialized on %s", device)
 
 
 
47
 
48
  # Define Pydantic model for slide data
49
  class Slide(BaseModel):
 
1142
  )
1143
 
1144
  if __name__ == "__main__":
1145
+ demo.launch(allowed_paths=[OUTPUT_DIR])