Updated app.py

app.py

@@ -1,9 +1,9 @@
-from transformers import AutoModelWithLMHead, AutoTokenizer
+from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 import gradio as gr
 
-tokenizer = AutoTokenizer.from_pretrained('microsoft/DialoGPT-medium')
-model = AutoModelWithLMHead.from_pretrained('output-medium')
+tokenizer = AutoTokenizer.from_pretrained("natdon/DialoGPT_Michael_Scott", use_auth_token=True)
+model = AutoModelForCausalLM.from_pretrained("natdon/DialoGPT_Michael_Scott", use_auth_token=True)
 
 chat_history_ids = None
 step = 0