Update app.py

app.py

@@ -4,12 +4,11 @@ import threading
 from threading import Event
 from typing import Optional
 
-import asyncio
 import datetime
 import requests
 import discord
 import gradio as gr
-import gradio_client as grc
+from gradio_client import Client
 from discord import Permissions
 from discord.ext import commands
 from discord.utils import oauth_url
@@ -45,38 +44,17 @@ async def uptime(ctx):
 @bot.command()
 async def ai(ctx, *, input_text: str):
     """Ask our AI model a question. (Session resets every 1 message!)"""
-    
-    try:
-        client = grc.Client("https://wop-xxx-opengpt.hf.space/")
-        
-        # Remove leading and trailing whitespace from the input text
-        input_text = input_text.strip()
-        
-        # Asynchronously make the prediction request
-        prediction_task = asyncio.create_task(client.predict(
-            input_text,
-            0.9,
-            1800,
-            0.9,
-            1.2,
-            api_name="/chat"
-        ))
+    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]
         
-        # Wait for the prediction task to complete
-        result = await prediction_task
-        
-        # Truncate the last 4 characters from the result (if necessary)
-        result = result[:-4]
-        
-        # Create an embed with the AI's response
-        embed = discord.Embed(title="AI Response", description=result, color=discord.Color.green())
-        embed.add_field(name="Info", value="Session resets every `1` message!\nModel Name: `Mistral 7B`", inline=False)
-        embed.set_footer(text="Created by Cosmos")
+    # Create an embed with the AI's response
+    embed = discord.Embed(title="AI Response", description=result, color=discord.Color.green())
+    embed.add_field(name="Info", value="Session resets every `1` message!\nModel Name: `Mistral 7B`", inline=False)
+    embed.set_footer(text="Created by Cosmos")
         
-        # Reply with the embed
-        await ctx.send(embed=embed)
-    except Exception as e:
-        await ctx.send(f"An error occurred: {str(e)}")
+    # Reply with the embed
+    await ctx.send(embed=embed)
 
 
 @bot.command()