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,
|