Update app.py

app.py

@@ -6,14 +6,16 @@ Automatically generated by Colaboratory.
 """
 import os
 import gradio as gr
+import transformers
 from PIL import Image
 from timeit import default_timer as timer
 from tensorflow import keras
+from transformers import GPT2Tokenizer, GPT2LMHeadModel
 from transformers import AutoModelWithLMHead, AutoTokenizer
 import numpy as np
 
-loaded_model = AutoModelWithLMHead.from_pretrained('runaksh/medquad-finetuned-gpt2')
-loaded_tokenizer = AutoTokenizer.from_pretrained('runaksh/medquad-finetuned-gpt2')
+loaded_model = GPT2LMHeadModel.from_pretrained("gpt2")
+loaded_tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
 
 def generate_query_response(prompt, max_length=200):