Upload app.py

app.py

@@ -43,7 +43,7 @@ class CNN(nn.Module):
 
 # Load the trained model
 model = CNN(num_classes=5)
-model.load_state_dict(torch.load("best_model.pth", map_location="cpu"))
+model.load_state_dict(torch.load("best_model_CNN_bs32_lr0.0005_epoch9_acc0.9238.pth", map_location="cpu"))
 model.eval()
 
 # Extract features from audio file
@@ -71,11 +71,11 @@ def predict_and_reply(audio_path):
     prompt = f"The user sounds {emotion.lower()}. What would you like to say to them?"
 
     try:
-        openai.api_key = os.getenv("OPENAI_API_KEY", "your-openai-api-key")  # Replace with real key or env var
+        openai.api_key = os.getenv("OPENAI_API_KEY", "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")  # Replace with real key or env var
         response = openai.ChatCompletion.create(
             model="gpt-3.5-turbo",
             messages=[
-                {"role": "system", "content": "You are an empathetic AI assistant."},
+                {"role": "system", "content": "You are a helpful assistant that provides entertainment recommendations."},
                 {"role": "user", "content": prompt}
             ]
         )