Update app.py

app.py

@@ -28,7 +28,7 @@ except requests.exceptions.RequestException as e:
 
 def generate_story(prompt, style):
     try:
-        client = InferenceClient(api_key=get_random_api_key(), model_id="Qwen/QwQ-32B-Preview")
+        client = InferenceClient(api_key=get_random_api_key())
         
         language = detect(prompt)
         if language != 'en':
@@ -39,7 +39,7 @@ def generate_story(prompt, style):
             {"role": "user", "content": prompt}
         ]
 
-        completion = client.chat.completions.create(messages=messages, max_tokens=1000)
+        completion = client.chat.completions.create(model="Qwen/QwQ-32B-Preview", messages=messages, max_tokens=1000)
         story = completion.choices[0].message.content
         return story
     except Exception as e:
@@ -48,7 +48,7 @@ def generate_story(prompt, style):
 
 def edit_story(original_story, edited_prompt):
     try:
-        client = InferenceClient(api_key=get_random_api_key(), model_id="Qwen/QwQ-32B-Preview")
+        client = InferenceClient(api_key=get_random_api_key())
         
         messages = [
             {"role": "system", "content": "Отредактируй историю, учитывая предоставленные указания."},
@@ -56,7 +56,7 @@ def edit_story(original_story, edited_prompt):
             {"role": "assistant", "content": original_story}
         ]
         
-        completion = client.chat.completions.create(messages=messages, max_tokens=1000)
+        completion = client.chat.completions.create(model="Qwen/QwQ-32B-Preview", messages=messages, max_tokens=1000)
         edited_story = completion.choices[0].message.content
         return edited_story
     except Exception as e: