satyamr196 commited on
Commit
acec6fc
·
1 Parent(s): 0cb2c03

corrected status route and variable job_status

Browse files
Files changed (1) hide show
  1. ASR_Server.py +7 -3
ASR_Server.py CHANGED
@@ -7,7 +7,12 @@ import threading
7
  import os
8
  os.environ["HF_HOME"] = "/tmp/huggingface"
9
 
10
-
 
 
 
 
 
11
  csv_path = "test.csv"
12
  df = pd.read_csv(csv_path)
13
  print(f"CSV Loaded with {len(df)} rows")
@@ -91,7 +96,6 @@ def generateTranscript(ASR_model, csv_path, output_dir="./"):
91
  import soundfile as sf
92
  from transformers import pipeline
93
 
94
- global job_status
95
  job_status.update({
96
  "running": True,
97
  "model": ASR_model,
@@ -224,7 +228,7 @@ def asr_models():
224
  print("Transcription started in background")
225
  return jsonify({"asr_models": models})
226
 
227
- @app.route("/job_status")
228
  def get_status():
229
  return jsonify(job_status)
230
 
 
7
  import os
8
  os.environ["HF_HOME"] = "/tmp/huggingface"
9
 
10
+ job_status = {
11
+ "running": False,
12
+ "model": None,
13
+ "completed": 0,
14
+ "message": "No job running"
15
+ }
16
  csv_path = "test.csv"
17
  df = pd.read_csv(csv_path)
18
  print(f"CSV Loaded with {len(df)} rows")
 
96
  import soundfile as sf
97
  from transformers import pipeline
98
 
 
99
  job_status.update({
100
  "running": True,
101
  "model": ASR_model,
 
228
  print("Transcription started in background")
229
  return jsonify({"asr_models": models})
230
 
231
+ @app.route("/status")
232
  def get_status():
233
  return jsonify(job_status)
234