Update app.py

app.py

@@ -1,16 +1,12 @@
 import time
 from transformers import pipeline, Conversation, GPT2LMHeadModel, AutoTokenizer
 import gradio as gr
-import torch
 
 model_name = "ingen51/DialoGPT-medium-GPT4"
 
 model = GPT2LMHeadModel.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name, padding_side='left')
 
-device = torch.device("cuda")
-model.to(device)
-
 chatbot = pipeline(task="conversational", model = model, tokenizer = tokenizer)
 
 past_user_inputs = []