Update app.py

app.py

@@ -1,53 +1,6 @@
-# import gradio as gr
-# from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
-
-# tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
-# model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")
-
-# def chat_with_model(input_text):
-#     input_ids = tokenizer.encode("You: " + input_text, return_tensors="pt", max_length=512, truncation=True)
-#     response_ids = model.generate(input_ids, max_length=100, num_return_sequences=1, no_repeat_ngram_size=2)
-#     reply = tokenizer.decode(response_ids[0], skip_special_tokens=True)
-#     return reply
-
-# iface = gr.Interface(
-#     fn=chat_with_model,
-#     inputs=gr.Textbox(prompt="You:"),
-#     outputs=gr.Textbox(prompt="Bot:"),
-#     title="Hello Mate!! 😁" ,
-# )
-
-# iface.launch()
-
-
-
-
 import gradio as gr
 from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
 
-DESCRIPTION = "This Application is developed just for Education purpose"
-LICENSE = "All Rights are Reserved to the Owner, Regards Mkoot"
-custom_css = """body {
-    background-image: url('BK.jpg');
-    background-size: cover;
-}
-
-input[type="text"] {
-    background-color: #f2f2f2; 
-    color: #333; 
-    border: 1px solid #ccc; 
-    border-radius: 5px; 
-}
-
-button {
-    background-color: #007bff; 
-    color: #fff; 
-    border: none; 
-    border-radius: 5px; 
-    padding: 10px 20px; 
-    cursor: pointer;
-}"""
-
 tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
 model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")
 
@@ -57,16 +10,15 @@ def chat_with_model(input_text):
     reply = tokenizer.decode(response_ids[0], skip_special_tokens=True)
     return reply
 
-with gr.Blocks(css=custom_css) as custom_css:
-    gr.Markdown(DESCRIPTION)
-    chat_interface = gr.Interface(
-        fn=chat_with_model,
-        inputs=gr.Textbox(prompt="You:"),
-        outputs=gr.Textbox(prompt="Bot:"),
-        title="Hello Mate!! 😁",
-    )
-    chat_interface.launch(share=True)
-    gr.Markdown(LICENSE)
+iface = gr.Interface(
+    fn=chat_with_model,
+    inputs=gr.Textbox(prompt="You:"),
+    outputs=gr.Textbox(prompt="Bot:"),
+    title="Hello Mate!! 😁" ,
+)
+
+iface.launch()
+