Update main.py

main.py

@@ -45,8 +45,9 @@ def generate(item: Item):
         seed=item.seed,
     )
 
-    if not client:
+    if client is None:
         client = InferenceClient(token=item.token, model=model_id)
+        
     formatted_prompt = format_prompt(item)
     stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""