semakoc commited on
Commit
a08d409
·
verified ·
1 Parent(s): a95859c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -7,12 +7,15 @@ import os
7
  import firebase_admin
8
  from firebase_admin import credentials, firestore
9
  from datetime import datetime
 
10
 
11
  # Initialize Firebase
 
12
  cred = credentials.Certificate('firebase_credentials.json') # Your Firebase JSON key file
13
  firebase_admin.initialize_app(cred)
14
  db = firestore.client()
15
 
 
16
  # Load the ASR model and processor
17
  MODEL_NAME = "eleferrand/xlsr53_Amis"
18
  processor = Wav2Vec2Processor.from_pretrained(MODEL_NAME)
@@ -57,9 +60,9 @@ def store_correction(original_transcription, corrected_transcription):
57
  'timestamp': datetime.now().isoformat()
58
  }
59
  db.collection('transcription_corrections').add(correction_data)
60
- return "Correction saved successfully!"
61
  except Exception as e:
62
- return f"⚠️ Error saving correction: {e}"
63
 
64
  def prepare_download(audio_file, original_transcription, corrected_transcription):
65
  """
 
7
  import firebase_admin
8
  from firebase_admin import credentials, firestore
9
  from datetime import datetime
10
+ import json
11
 
12
  # Initialize Firebase
13
+ firebase_config = json.loads(os.environ.get('firebase_creds')
14
  cred = credentials.Certificate('firebase_credentials.json') # Your Firebase JSON key file
15
  firebase_admin.initialize_app(cred)
16
  db = firestore.client()
17
 
18
+
19
  # Load the ASR model and processor
20
  MODEL_NAME = "eleferrand/xlsr53_Amis"
21
  processor = Wav2Vec2Processor.from_pretrained(MODEL_NAME)
 
60
  'timestamp': datetime.now().isoformat()
61
  }
62
  db.collection('transcription_corrections').add(correction_data)
63
+ return "Correction saved successfully!"
64
  except Exception as e:
65
+ return f"Error saving correction: {e}"
66
 
67
  def prepare_download(audio_file, original_transcription, corrected_transcription):
68
  """