Update app.py

app.py

@@ -1,3 +1,8 @@
+import gradio as gr
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
+
+
 title = "Ask Rick a Question"
 description = """
 <center>
@@ -8,9 +13,6 @@ The bot was trained to answer questions based on Rick and Morty dialogues. Ask R
 
 article = "Check out [the original Rick and Morty Bot](https://huggingface.co/spaces/kingabzpro/Rick_and_Morty_Bot) that this demo is based off of."
 
-from transformers import AutoModelForCausalLM, AutoTokenizer
-import torch
-
 tokenizer = AutoTokenizer.from_pretrained("ericzhou/DialoGPT-Medium-Rick_v2")
 model = AutoModelForCausalLM.from_pretrained("ericzhou/DialoGPT-Medium-Rick_v2")