Update app.py

app.py

@@ -18,7 +18,12 @@ event = Event()
 
 DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
 HF_TOKEN = os.getenv("HF_TOKEN")
-
+def ask(input_text):
+    client = Client("https://wop-xxx-opengpt.hf.space/")
+    result = client.predict(input_text, 0.9, 2000, 0.9, 1.2, api_name="/chat")
+    result = result[:-4]
+    return result
+    
 async def wait(job):
     while not job.done():
         await asyncio.sleep(0.2)
@@ -44,9 +49,7 @@ async def uptime(ctx):
 @bot.command()
 async def ai(ctx, *, input_text: str):
     """Ask our AI model a question. (Session resets every 1 message!)"""
-    client = Client("https://wop-xxx-opengpt.hf.space/")
-    result = client.predict(input_text, 0.9, 2000, 0.9, 1.2, api_name="/chat")
-    result = result[:-4]
+    result = ask(input_text)
         
     # Create an embed with the AI's response
     embed = discord.Embed(title="AI Response", description=result, color=discord.Color.green())