Update app.py
Browse files
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 "
|
61 |
except Exception as e:
|
62 |
-
return f"
|
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 |
"""
|