Update app.py

app.py

@@ -4,6 +4,8 @@ import gradio as gr
 from transformers import pipeline, Conversation
 
 model = "mistralai/Mistral-7B-Instruct-v0.1"
+
+# Test Model
 model = "TinyLlama/TinyLlama-1.1B-Chat-v0.3"
 
 title = "Shisa 7B"
@@ -19,12 +21,6 @@ examples = [
 # Docs: https://github.com/huggingface/transformers/blob/main/src/transformers/pipelines/conversational.py
 conversation = Conversation()
 chatbot = pipeline('conversational', model)
-'''
-conversation = Conversation("Going to the movies tonight - any suggestions?")
-conversation.add_message({"role": "assistant", "content": "The Big lebowski."})
-conversation.add_message({"role": "user", "content": "Is it good?"})
-conversation.messages[:-1]
-'''
 
 def chat(input, history=[]):
     conversation.add_message({"role": "user", "content": input})
@@ -51,29 +47,5 @@ gr.ChatInterface(
     clear_btn="Clear",
 ).launch()
 
-'''
-gr.Interface.load(
-    "EleutherAI/gpt-j-6B",
-    inputs=gr.Textbox(lines=5, label="Input Text"),
-    title=title,
-    description=description,
-    article=article,
-).launch()
-
-
-# Doesn't support conversational pipelin
-pipe = pipeline('conversational', model)
-gr.Interface.from_pipeline(pipe).launch()
-
-
-'''
-
-
 # For async
-# ).queue().launch()
-
-'''
-# Pipeline doesn't support conversational...
-pipe = pipeline("conversational", model=model)
-demo = gr.Interface.from_pipeline(pipe)
-'''
+# ).queue().launch()