Update app.py

app.py

@@ -1,8 +1,8 @@
 import transformers
 import gradio as gr
-import warnings
+# import warnings
 import torch
-warnings.simplefilter('ignore')
+# warnings.simplefilter('ignore')
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -16,11 +16,11 @@ tokenizer.add_special_tokens(
 })
 #add bot token since it is not a special token
 tokenizer.add_tokens(["<bot>:"])
-
+print("=====Done 1")
 model = transformers.GPT2LMHeadModel.from_pretrained('gpt2')
 model.resize_token_embeddings(len(tokenizer))
-model.load_state_dict(torch.load('gpt2talk.pt', map_location=torch.device('cpu')))
-
+model.load_state_dict(torch.load('./gpt2talk.pt', map_location=torch.device('cpu')))
+print("=====Done 2")
 model.eval()
 def inference(quiz):
     quiz1 = quiz
@@ -40,6 +40,7 @@ def chatbot(input_text):
     return response
 
 # Create the Gradio interface
+print("=====Done 3")
 iface = gr.Interface(
     fn=chatbot,
     inputs=gr.Textbox(),
@@ -49,6 +50,6 @@ iface = gr.Interface(
     title="ChatFinance",
     description="Ask the a question and see its response!",
 )
-
+print("=====Done 4")
 # Launch the Gradio interface
 iface.launch()