futranbg commited on
Commit
dddc929
1 Parent(s): aac3374

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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)