acecalisto3 commited on
Commit
5edaad8
·
verified ·
1 Parent(s): ce328d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -221,7 +221,7 @@ class Model:
221
  self.model_link = model_link
222
  self.inference_client = InferenceClient(model=model_link)
223
 
224
- def generate_text(self, prompt, temperature=0.5, max_new_tokens=2048):
225
  try:
226
  output = self.inference_client.text_generation(
227
  prompt,
 
221
  self.model_link = model_link
222
  self.inference_client = InferenceClient(model=model_link)
223
 
224
+ def generate_text(self, prompt, temperature=0.5, max_new_tokens=4096):
225
  try:
226
  output = self.inference_client.text_generation(
227
  prompt,