diff --git a/lulzbot.py b/lulzbot.py
index ab6d026..81cc2a7 100644
--- a/lulzbot.py
+++ b/lulzbot.py
@@ -22,7 +22,7 @@ import torch
 import asyncio
 
 tokenizer = AutoTokenizer.from_pretrained('microsoft/DialoGPT-large')
-model = AutoModelForCausalLM.from_pretrained('../southpark/output-small')
+model = AutoModelForCausalLM.from_pretrained('../southpark/output-medium')
 
 # Twitter
 import tweepy