hatimanees commited on
Commit
7f13e81
·
verified ·
1 Parent(s): 1f56d49

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +6 -5
run.py CHANGED
@@ -7,8 +7,6 @@ import torch
7
  import re
8
  from werkzeug.utils import secure_filename
9
  import uuid
10
-
11
- import os
12
  import platform
13
 
14
  # Set Transformers Cache Directory
@@ -35,7 +33,7 @@ print(f"Environment variable TRANSFORMERS_CACHE set to '{transformers_cache_dire
35
 
36
 
37
  class Config:
38
- UPLOAD_FOLDER = os.path.join(os.path.dirname(__file__), 'uploads') # Correct path
39
  MAX_CONTENT_LENGTH = 16 * 1024 * 1024 # 16MB max file size
40
  CORS_HEADERS = 'Content-Type'
41
 
@@ -148,10 +146,12 @@ def analyze_sentiment(file_path: str):
148
  return [{'label': 'Error', 'score': 0.5}]
149
 
150
 
 
151
  def create_app():
152
  app = Flask(__name__)
153
  app.config.from_object(Config)
154
- CORS(app)
 
155
  os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
156
 
157
  @app.route('/upload', methods=['POST'])
@@ -162,7 +162,7 @@ def create_app():
162
  return jsonify({'error': 'No transcript received'}), 400
163
 
164
  # Save the transcript in the current folder
165
- file_path = os.path.join(app.config['UPLOAD_FOLDER'], 'transcript.txt')
166
  with open(file_path, 'w') as file:
167
  file.write(transcript)
168
 
@@ -180,6 +180,7 @@ def create_app():
180
 
181
 
182
 
 
183
  if __name__ == '__main__':
184
  app = create_app()
185
  app.run(host="0.0.0.0", port=5000)
 
7
  import re
8
  from werkzeug.utils import secure_filename
9
  import uuid
 
 
10
  import platform
11
 
12
  # Set Transformers Cache Directory
 
33
 
34
 
35
  class Config:
36
+ UPLOAD_FOLDER = os.path.join(os.path.dirname(__file__), '/tmp/uploads') # Correct path
37
  MAX_CONTENT_LENGTH = 16 * 1024 * 1024 # 16MB max file size
38
  CORS_HEADERS = 'Content-Type'
39
 
 
146
  return [{'label': 'Error', 'score': 0.5}]
147
 
148
 
149
+
150
  def create_app():
151
  app = Flask(__name__)
152
  app.config.from_object(Config)
153
+
154
+ # Ensure the uploads directory exists
155
  os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
156
 
157
  @app.route('/upload', methods=['POST'])
 
162
  return jsonify({'error': 'No transcript received'}), 400
163
 
164
  # Save the transcript in the current folder
165
+ file_path = os.path.join(os.getcwd(), 'transcript.txt')
166
  with open(file_path, 'w') as file:
167
  file.write(transcript)
168
 
 
180
 
181
 
182
 
183
+
184
  if __name__ == '__main__':
185
  app = create_app()
186
  app.run(host="0.0.0.0", port=5000)