Error fixed?!

app.py

@@ -1,18 +1,27 @@
 import gradio as gr
-import random
-import time
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
-ith gr.Blocks() as demo:
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox()
-    clear = gr.ClearButton([msg, chatbot])
+# Load the DialoGPT model and tokenizer
+model_name = "models/Astrea/DialoGPT-small-akari"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
 
-    def respond(message, chat_history):
-        bot_message = random.choice(["How are you?", "I love you", "I'm very hungry"])
-        chat_history.append((message, bot_message))
-        time.sleep(2)
-        return "", chat_history
+# Define the Gradio interface
+akari = gr.Interface(
+    fn=lambda messages: {
+        "response": model.generate(
+            tokenizer.encode(messages["input"], return_tensors="pt"),
+            max_length=50,
+            num_beams=5,
+            no_repeat_ngram_size=2,
+            top_k=50,
+            top_p=0.95,
+            temperature=0.7,
+        )[0].decode("utf-8")
+    },
+    inputs=gr.Textbox(),
+    outputs=gr.Textbox(),
+)
 
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
-
-gr.load("models/Astrea/DialoGPT-small-akari").launch()
+akari.launch()