Update README.md

README.md

@@ -66,3 +66,14 @@ from huggingface_hub import notebook_login
 notebook_login()
 model.push_to_hub("your-huggingface-username/custom-gpt")
 tokenizer.push_to_hub("your-huggingface-username/custom-gpt")
+pip install gradio
+import gradio as gr
+from transformers import pipeline
+
+generator = pipeline("text-generation", model="your-huggingface-username/custom-gpt")
+
+def chatbot(prompt):
+    return generator(prompt, max_length=100)[0]["generated_text"]
+
+iface = gr.Interface(fn=chatbot, inputs="text", outputs="text")
+iface.launch()