Phong1807 commited on
Commit
c31a4e8
1 Parent(s): d24a565

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -10
app.py CHANGED
@@ -1,8 +1,7 @@
1
  from flask import Flask, request, jsonify
2
  from flask_cors import CORS, cross_origin
3
  from transformers import pipeline
4
- import firebase_admin
5
- from firebase_admin import credentials, firestore
6
  import os
7
 
8
  # Kiểm tra biến môi trường TRANSFORMERS_CACHE
@@ -16,20 +15,12 @@ app.config['CORS_HEADERS'] = 'Content-type'
16
  # Tải mô hình Hugging Face
17
  transcriber = pipeline(model="Phong1807/whisper-datn11")
18
 
19
- # Thiết lập Firebase
20
- cred = credentials.Certificate('whisper-1da8a-firebase-adminsdk-1bfxq-fe43b4015d.json')
21
- firebase_admin.initialize_app(cred)
22
- db = firestore.client()
23
-
24
  @app.route('/transcribe', methods=['POST'])
25
  @cross_origin(origin='*')
26
  def transcribe():
27
  audio_data = request.files['audio']
28
  transcription = transcriber(audio_data)['text']
29
 
30
- # Upload transcription lên Firebase
31
- doc_ref = db.collection('transcriptions').add({'text': transcription})
32
-
33
  return jsonify({'transcription': transcription})
34
 
35
  if __name__ == '__main__':
 
1
  from flask import Flask, request, jsonify
2
  from flask_cors import CORS, cross_origin
3
  from transformers import pipeline
4
+
 
5
  import os
6
 
7
  # Kiểm tra biến môi trường TRANSFORMERS_CACHE
 
15
  # Tải mô hình Hugging Face
16
  transcriber = pipeline(model="Phong1807/whisper-datn11")
17
 
 
 
 
 
 
18
  @app.route('/transcribe', methods=['POST'])
19
  @cross_origin(origin='*')
20
  def transcribe():
21
  audio_data = request.files['audio']
22
  transcription = transcriber(audio_data)['text']
23
 
 
 
 
24
  return jsonify({'transcription': transcription})
25
 
26
  if __name__ == '__main__':