Sharathhebbar24 commited on
Commit
e704df6
1 Parent(s): 60dab54

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +6 -6
main.py CHANGED
@@ -11,11 +11,11 @@ class UserInterface():
11
  st.text("An Open Source Chat Application")
12
  st.header("Open LLMs")
13
 
14
- self.API_KEY = st.sidebar.text_input(
15
- 'API Key',
16
- type='password',
17
- help="Type in your HuggingFace API key to use this app"
18
- )
19
 
20
  models_name = (
21
  "HuggingFaceH4/zephyr-7b-beta",
@@ -51,7 +51,7 @@ class UserInterface():
51
  "max_length": self.max_token_length
52
  }
53
 
54
- os.environ['HUGGINGFACEHUB_API_TOKEN'] = self.API_KEY
55
 
56
 
57
  def form_data(self):
 
11
  st.text("An Open Source Chat Application")
12
  st.header("Open LLMs")
13
 
14
+ # self.API_KEY = st.sidebar.text_input(
15
+ # 'API Key',
16
+ # type='password',
17
+ # help="Type in your HuggingFace API key to use this app"
18
+ # )
19
 
20
  models_name = (
21
  "HuggingFaceH4/zephyr-7b-beta",
 
51
  "max_length": self.max_token_length
52
  }
53
 
54
+ os.environ['HUGGINGFACEHUB_API_TOKEN'] = HF_KEY
55
 
56
 
57
  def form_data(self):