Update app.py

app.py

@@ -9,6 +9,9 @@ import openai
 from torch import nn
 import torch.nn.functional as F
 from nltk.tokenize import sent_tokenize
+import os
+
+print(os.listdir('.'))
 
 # Download the 'punkt' package
 nltk.download('punkt')
@@ -35,7 +38,7 @@ class Bert_model(nn.Module):
         return logits
 
 # Load the model
-model_path = "Jforeverss/finchat222/model.pt"  # Replace with your actual model path
+model_path = "finchat222/model.pt"  # Replace with your actual model path
 state_dict = torch.load(model_path)
 device = "cuda"  # or "cpu" if GPU is not available