edited app file

app.py

@@ -1,12 +1,11 @@
 import gradio as gr
 from transformers import GPT2Tokenizer, GPT2LMHeadModel, TextDataset, DataCollatorForLanguageModeling
 from transformers import Trainer, TrainingArguments
+from os.path import dirname
 
 
-
-model_output_path="/repo"
-my_model = GPT2LMHeadModel.from_pretrained(model_output_path)
-my_tokenizer = GPT2Tokenizer.from_pretrained(model_output_path)
+my_model = GPT2LMHeadModel.from_pretrained(f'{dirname(__file__)}/repo')
+my_tokenizer = GPT2Tokenizer.from_pretrained(f'{dirname(__file__)}/repo')
 
 def predict(prompt):
   response = generate_response(my_model, my_tokenizer, prompt)