Bui commited on
Commit
07fede4
1 Parent(s): 8800f06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -2,14 +2,18 @@ from flask import Flask, request, jsonify
2
  from transformers import pipeline
3
  import firebase_admin
4
  from firebase_admin import credentials, firestore
 
 
 
 
5
 
6
  app = Flask(__name__)
7
 
8
  # Tải mô hình Hugging Face
9
- transcriber = pipeline(model="Dung125/whisper-datn11")
10
 
11
  # Thiết lập Firebase
12
- cred = credentials.Certificate('whisper-1da8a-firebase-adminsdk-1bfxq-fe43b4015d.json')
13
  firebase_admin.initialize_app(cred)
14
  db = firestore.client()
15
 
 
2
  from transformers import pipeline
3
  import firebase_admin
4
  from firebase_admin import credentials, firestore
5
+ import os
6
+
7
+ # Kiểm tra biến môi trường TRANSFORMERS_CACHE
8
+ print(f"TRANSFORMERS_CACHE: {os.getenv('TRANSFORMERS_CACHE')}")
9
 
10
  app = Flask(__name__)
11
 
12
  # Tải mô hình Hugging Face
13
+ transcriber = pipeline('automatic-speech-recognition', model='your-username/your-model-name')
14
 
15
  # Thiết lập Firebase
16
+ cred = credentials.Certificate('path/to/your/firebase/credentials.json')
17
  firebase_admin.initialize_app(cred)
18
  db = firestore.client()
19