rishisim commited on
Commit
b361cc6
·
verified ·
1 Parent(s): 5fe5f37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -12,21 +12,21 @@ import os
12
 
13
 
14
  # from transformers import AutoModelForCausalLM, AutoTokenizer
15
- from transformers import pipeline
16
 
17
- hf_token = os.environ['llama_token']
18
 
19
- login(token=hf_token)
20
 
21
- llm = pipeline("text-generation", model="meta-llama/Meta-Llama-3-8B-Instruct")
22
 
23
  # llm = pipeline("text-generation", model = "meta-llama/Meta-Llama-3-70B-Instruct")
24
 
25
  # llm = pipeline("text-generation", model="mistralai/Mixtral-8x22B-Instruct-v0.1")
26
 
27
- # api_key = "AIzaSyCdM_aAIsW_nPbjarOF83mbX1_z1cVX2_M"
28
 
29
- # llm = GoogleGenerativeAI(model="models/text-bison-001", google_api_key=api_key)
30
  # llm = GooglePalm(google_api_key = api_key, temperature=0.7)
31
 
32
  # LOADING CSV FILE
 
12
 
13
 
14
  # from transformers import AutoModelForCausalLM, AutoTokenizer
15
+ # from transformers import pipeline
16
 
17
+ # hf_token = os.environ['llama_token']
18
 
19
+ # login(token=hf_token)
20
 
21
+ # llm = pipeline("text-generation", model="meta-llama/Meta-Llama-3-8B-Instruct")
22
 
23
  # llm = pipeline("text-generation", model = "meta-llama/Meta-Llama-3-70B-Instruct")
24
 
25
  # llm = pipeline("text-generation", model="mistralai/Mixtral-8x22B-Instruct-v0.1")
26
 
27
+ api_key = "AIzaSyCdM_aAIsW_nPbjarOF83mbX1_z1cVX2_M"
28
 
29
+ llm = GoogleGenerativeAI(model="models/text-bison-001", google_api_key=api_key)
30
  # llm = GooglePalm(google_api_key = api_key, temperature=0.7)
31
 
32
  # LOADING CSV FILE