Ramaravind commited on
Commit
62cc68c
·
verified ·
1 Parent(s): 3ea3319

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,11 +4,12 @@ import os
4
 
5
  HF_API_TOKEN = os.getenv("HF_API_TOKEN")
6
  MODEL = "bigscience/bloom-560m" #"microsoft/Phi-3-mini-128k-instruct"
 
7
  # Initialize the InferenceClient with the token and model.
8
  # Using a lightweight model like google/flan-t5-small.
9
  client = InferenceClient(
10
  "google/flan-t5-small",
11
- token=HF_TOKEN
12
  )
13
 
14
 
 
4
 
5
  HF_API_TOKEN = os.getenv("HF_API_TOKEN")
6
  MODEL = "bigscience/bloom-560m" #"microsoft/Phi-3-mini-128k-instruct"
7
+
8
  # Initialize the InferenceClient with the token and model.
9
  # Using a lightweight model like google/flan-t5-small.
10
  client = InferenceClient(
11
  "google/flan-t5-small",
12
+ token=HF_API_TOKEN
13
  )
14
 
15