KaizeShi commited on
Commit
96c33c2
·
1 Parent(s): 262d1c8

Add application file

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ CONTEXT_LENGTH = int(os.environ.get("CONTEXT_LENGTH"))
17
  COLOR = os.environ.get("COLOR")
18
  DESCRIPTION = os.environ.get("DESCRIPTION")
19
  LORA_WEIGHTS = "DSMI/LLaMA-E"
20
-
21
 
22
  @spaces.GPU(duration=120)
23
  def predict(message, history, system_prompt, temperature, max_new_tokens, top_k, repetition_penalty, top_p):
 
17
  COLOR = os.environ.get("COLOR")
18
  DESCRIPTION = os.environ.get("DESCRIPTION")
19
  LORA_WEIGHTS = "DSMI/LLaMA-E"
20
+ access_token = os.environ.get('HF_TOKEN')
21
 
22
  @spaces.GPU(duration=120)
23
  def predict(message, history, system_prompt, temperature, max_new_tokens, top_k, repetition_penalty, top_p):