Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -34,22 +34,16 @@ logging.basicConfig(
|
|
34 |
logger = logging.getLogger(__name__)
|
35 |
|
36 |
# Set up environment
|
37 |
-
|
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 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
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]
|
|
|
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])
|