Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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=
|
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,
|