Update app.py

app.py

@@ -9,6 +9,33 @@ Original file is located at
 
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
+print("Starting app.py")
+
+import gradio as gr
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+print("Imports successful")
+
+model_name = "microsoft/DialoGPT-small"
+print(f"Loading model {model_name}...")
+
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+model.eval()
+
+print("Model loaded successfully")
+
+# your rest of the code...
+
+with gr.Blocks() as demo:
+    # ... UI components ...
+
+demo.launch()
+
+
+
+
+
 
 model_name = "microsoft/DialoGPT-small"
 tokenizer = AutoTokenizer.from_pretrained(model_name)