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])
|