Update app.py

app.py

@@ -2,10 +2,14 @@ import gradio as gr
 import torch
 from transformers import GPT2Tokenizer, GPT2LMHeadModel, TextDataset, DataCollatorForLanguageModeling
 from transformers import Trainer, TrainingArguments
+import numpy as np
+import pandas as pd
+from sklearn.model_selection import train_test_split
 
 __checkpoint = "gpt2"
 __tokenizer = GPT2Tokenizer.from_pretrained(__checkpoint)
 __model = GPT2LMHeadModel.from_pretrained(__checkpoint)
+__model_output_path = "/paavansundar/Models"
 # Create a Data collator object
 data_collator = DataCollatorForLanguageModeling(tokenizer=__tokenizer, mlm=False, return_tensors="pt")
 def queryGPT(question):