Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
-from transformers import pipeline
 from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
 
 tokenizer = AutoTokenizer.from_pretrained("sapientinc/HRM-Text-1B")
 model = AutoModelForCausalLM.from_pretrained("sapientinc/HRM-Text-1B")
@@ -15,8 +15,4 @@ def chat(message, history):
     
     return response
 
-with gr.Blocks(title="AI Queen") as demo:
-    gr.Markdown("# AI Queen")
-    gr.ChatInterface(fn=chat)
-
-demo.launch()
+gr.ChatInterface(fn=chat).launch()