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