Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 from transformers import DistilGPT2Tokenizer, DistilGPT2LMHeadModel
 
-# Load model and tokenizer
 tokenizer = DistilGPT2Tokenizer.from_pretrained("distilgpt2")
 model = DistilGPT2LMHeadModel.from_pretrained("distilgpt2")
 
@@ -11,7 +10,6 @@ def generate_response(prompt):
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return response
 
-# Define Gradio interface
 iface = gr.Interface(
     fn=generate_response,
     inputs="text",
@@ -20,5 +18,4 @@ iface = gr.Interface(
     description="A chatbot powered by DistilGPT-2."
 )
 
-if __name__ == "__main__":
-    iface.launch()
+iface.launch()