broadfield-dev commited on
Commit
ccd0d6e
·
verified ·
1 Parent(s): 3e899a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,7 +13,8 @@ def call_llm(input_text):
13
  "max_length": 500,
14
  "num_return_sequences": 1,
15
  "temperature": 0.7,
16
- "top_k": 50
 
17
  }
18
 
19
  # Generate text
 
13
  "max_length": 500,
14
  "num_return_sequences": 1,
15
  "temperature": 0.7,
16
+ "top_k": 50,
17
+ "trust_remote_code"=True,
18
  }
19
 
20
  # Generate text