Update app.py

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Load the pre-trained model and tokenizer from the saved directory
-model_path = "blexus_pretrained_test"
+model_path = "Blexus/Quble_Test_Model_v1_Pretrain"
 tokenizer = GPT2Tokenizer.from_pretrained(model_path)
 model = GPT2LMHeadModel.from_pretrained(model_path).to(device)