Marcus Posey commited on
Commit
8b342f5
·
1 Parent(s): e2b978c

Add huggingface token

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -1,5 +1,7 @@
1
  from vllm import LLM, SamplingParams
2
  import gradio as gr
 
 
3
 
4
 
5
  class TextCompletion:
@@ -14,6 +16,9 @@ class TextCompletion:
14
 
15
 
16
  if __name__ == "__main__":
 
 
 
17
  model = LLM(
18
  model="mep296/llama-3-8b-rephrase-quality",
19
  tokenizer="meta-llama/Meta-Llama-3-8B",
 
1
  from vllm import LLM, SamplingParams
2
  import gradio as gr
3
+ import os
4
+ from huggingface_hub import login
5
 
6
 
7
  class TextCompletion:
 
16
 
17
 
18
  if __name__ == "__main__":
19
+ HF_TOKEN = os.getenv('HF_TOKEN')
20
+ login(token=HF_TOKEN)
21
+
22
  model = LLM(
23
  model="mep296/llama-3-8b-rephrase-quality",
24
  tokenizer="meta-llama/Meta-Llama-3-8B",