Update app.py

app.py

@@ -4,6 +4,9 @@ import torch.nn as nn
 import sys
 import os
 
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
 
 from models.audio_model import FrozenExtractor, LinearProjection, AGRU, PersonalityRegressor
@@ -197,4 +200,14 @@ with gr.Blocks(title="Personality Prediction API", theme=gr.themes.Soft()) as de
 
     gr.Markdown("---\n**Traits:** Extraversion · Agreeableness · Conscientiousness · Neuroticism · Openness")
 
-demo.launch()
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+app = gr.mount_gradio_app(app, demo, path="/")