Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -62,6 +62,7 @@ def generate(
|
|
| 62 |
if model in clientList:
|
| 63 |
client = InferenceClient(model, token=HF_TOKEN)
|
| 64 |
print(f"Choosen model: {model}")
|
|
|
|
| 65 |
|
| 66 |
try:
|
| 67 |
stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
|
|
|
|
| 62 |
if model in clientList:
|
| 63 |
client = InferenceClient(model, token=HF_TOKEN)
|
| 64 |
print(f"Choosen model: {model}")
|
| 65 |
+
break
|
| 66 |
|
| 67 |
try:
|
| 68 |
stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
|