Update app.py

app.py

@@ -1,8 +1,8 @@
 
-from transformers import AutoTokenizer, AutoModelForCausalLM
+#from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
 import torch
-#from transformers import GPT2LMHeadModel, GPT2Tokenizer
+from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
 #import pickle
 
@@ -12,12 +12,12 @@ description = "Building open-domain chatbots is a challenging area for machine l
 examples = [["Guvi Details"]]
 
 model_name = "fine_tuned_model123"
-#model = GPT2LMHeadModel.from_pretrained(model_name)
-#tokenizer = GPT2Tokenizer.from_pretrained(model_name)
+model = GPT2LMHeadModel.from_pretrained(model_name)
+tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 
 # Load the tokenizer and model from Hugging Face Hub
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name)
+#tokenizer = AutoTokenizer.from_pretrained(model_name)
+#model = AutoModelForCausalLM.from_pretrained(model_name)
 
 def predict(input, history=[]):
     # tokenize the new input sentence