Update app.py

app.py

@@ -26,9 +26,9 @@ except requests.exceptions.RequestException as e:
     css = " h1{text-align:center} .container { max-width: 800px; margin: 0 auto; }"
 
 
-def generate_story(prompt, style, api_key):
+def generate_story(prompt, style):
     try:
-        client = InferenceClient(hf_token=api_key, model_id="Qwen/QwQ-32B-Preview")
+        client = InferenceClient(api_key=get_random_api_key(), model_id="Qwen/QwQ-32B-Preview")
         
         language = detect(prompt)
         if language != 'en':
@@ -46,9 +46,9 @@ def generate_story(prompt, style, api_key):
         return f"Ошибка генерации: {e}"
 
 
-def edit_story(original_story, edited_prompt, api_key):
+def edit_story(original_story, edited_prompt):
     try:
-        client = InferenceClient(api_key=api_key, model_id="Qwen/QwQ-32B-Preview")
+        client = InferenceClient(api_key=get_random_api_key(), model_id="Qwen/QwQ-32B-Preview")
         
         messages = [
             {"role": "system", "content": "Отредактируй историю, учитывая предоставленные указания."},
@@ -79,8 +79,7 @@ with gr.Blocks(css=css) as demo:
         edited_story = gr.Textbox(label="Отредактированная история", lines=10)
         edit_button = gr.Button("Отредактировать", variant='primary')
     
-    api_key = get_random_api_key()
-    generate_button.click(generate_story, inputs=[prompt, style], outputs=[output_story], api_key=api_key)
+    generate_button.click(generate_story, inputs=[prompt, style], outputs=[output_story])
     edit_button.click(edit_story, inputs=[output_story, edited_prompt], outputs=[edited_story], api_key=api_key)
 
 demo.launch(show_api=False, share=False,concurrency_limit=250)