Add application file
Browse files
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):
|