Use conversational

app.py

@@ -4,7 +4,6 @@ import gradio
 history = []
 
 
-
 def get_history_messages():
     messages = []
     for user, assist in history:
@@ -14,8 +13,8 @@ def get_history_messages():
 
 
 def predict(prompt):
-    pipe = pipeline("text-generation", model="cognitivecomputations/TinyDolphin-2.8-1.1b")
-    response = pipe.generate(
+    pipe = pipeline("conversational", model="cognitivecomputations/TinyDolphin-2.8-1.1b")
+    response = pipe(
         [
             *get_history_messages(),
             {"role": "user", "content": prompt}
@@ -24,7 +23,7 @@ def predict(prompt):
     history.append((prompt, ""))
     message = ""
     for chunk in response:
-        message += chunk["message"]["content"]
+        message += chunk.messages[-1]["content"]
         history[-1] = (prompt, message)
         yield "", history