Paulie-Aditya commited on
Commit
296b6b4
·
1 Parent(s): 6cc8cb9

api issue fixed using env

Browse files
Files changed (2) hide show
  1. app.py +5 -2
  2. requirements.txt +1 -0
app.py CHANGED
@@ -5,6 +5,11 @@ import nltk
5
  from nltk import sent_tokenize
6
  from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
7
  from transformers import pipeline
 
 
 
 
 
8
 
9
  # Use a pipeline as a high-level helper
10
  from transformers import pipeline
@@ -15,8 +20,6 @@ tokenizer = MBart50TokenizerFast.from_pretrained("SnypzZz/Llama2-13b-Language-tr
15
  model = None
16
  model_loaded = False
17
 
18
- from secrets_file import api_token_header
19
-
20
 
21
  def load_model():
22
  global model, model_loaded
 
5
  from nltk import sent_tokenize
6
  from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
7
  from transformers import pipeline
8
+ import os
9
+ from dotenv import load_dotenv, dotenv_values
10
+ load_dotenv()
11
+
12
+ api_token_header = os.getenv("api_token_header")
13
 
14
  # Use a pipeline as a high-level helper
15
  from transformers import pipeline
 
20
  model = None
21
  model_loaded = False
22
 
 
 
23
 
24
  def load_model():
25
  global model, model_loaded
requirements.txt CHANGED
@@ -34,6 +34,7 @@ pydantic_core==2.27.2
34
  pydub==0.25.1
35
  Pygments==2.18.0
36
  python-dateutil==2.9.0.post0
 
37
  python-multipart==0.0.20
38
  pytz==2024.2
39
  PyYAML==6.0.2
 
34
  pydub==0.25.1
35
  Pygments==2.18.0
36
  python-dateutil==2.9.0.post0
37
+ python-dotenv==1.0.1
38
  python-multipart==0.0.20
39
  pytz==2024.2
40
  PyYAML==6.0.2