Update app.py

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 # Load the tokenizer and model
-tokenizer = AutoTokenizer.from_pretrained("sergeantson/GPT2_Small_Law_FT")
-model = AutoModelForCausalLM.from_pretrained("sergeantson/GPT2_Small_Law_FT")
+tokenizer = AutoTokenizer.from_pretrained("sergeantson/GPT2_Small_Law")
+model = AutoModelForCausalLM.from_pretrained("sergeantson/GPT2_Small_Law")
 
 def generate_text(input_text, max_length, num_return_sequences, temperature, top_k, top_p):
     inputs = tokenizer(input_text, return_tensors="pt")