hatimanees commited on
Commit
caa5a42
·
verified ·
1 Parent(s): c7c9cac

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +3 -2
run.py CHANGED
@@ -10,11 +10,12 @@ import uuid
10
 
11
 
12
  class Config:
13
- UPLOAD_FOLDER = os.path.join(os.path.dirname(__file__), 'uploads')
14
  MAX_CONTENT_LENGTH = 16 * 1024 * 1024 # 16MB max file size
15
  CORS_HEADERS = 'Content-Type'
16
 
17
 
 
18
  class DialogueSentimentAnalyzer:
19
  def __init__(self, model_name: str = "microsoft/DialogRPT-updown"):
20
  self.device = 0 if torch.cuda.is_available() else -1
@@ -136,7 +137,7 @@ def create_app():
136
  return jsonify({'error': 'No transcript received'}), 400
137
 
138
  # Save the transcript in the current folder
139
- file_path = os.path.join(os.getcwd(), 'transcript.txt')
140
  with open(file_path, 'w') as file:
141
  file.write(transcript)
142
 
 
10
 
11
 
12
  class Config:
13
+ UPLOAD_FOLDER = os.path.join(os.path.dirname(__file__), 'uploads') # Correct path
14
  MAX_CONTENT_LENGTH = 16 * 1024 * 1024 # 16MB max file size
15
  CORS_HEADERS = 'Content-Type'
16
 
17
 
18
+
19
  class DialogueSentimentAnalyzer:
20
  def __init__(self, model_name: str = "microsoft/DialogRPT-updown"):
21
  self.device = 0 if torch.cuda.is_available() else -1
 
137
  return jsonify({'error': 'No transcript received'}), 400
138
 
139
  # Save the transcript in the current folder
140
+ file_path = os.path.join(app.config['UPLOAD_FOLDER'], 'transcript.txt')
141
  with open(file_path, 'w') as file:
142
  file.write(transcript)
143