Switched to google/gemma-2b-it

app.py

@@ -5,8 +5,8 @@ import re
 
 # Load model and tokenizer
 print("Loading gemma...")
-tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
-model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
+tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b-it")
+model = AutoModelForCausalLM.from_pretrained("google/gemma-2b-it")
 
 # Add pad token if it doesn't exist
 if tokenizer.pad_token is None: