Update app.py

app.py

@@ -71,7 +71,10 @@ def generate(prompt, history,max_new_tokens,temperature=temperature,top_p=top_p,
 
     for response in stream:
         output += response.token.text
-        yield "",[history,(prompt,output)]
+        if history:
+            yield "",[history,[(prompt,output)]]
+        else:
+            yield "",[(prompt,output)]
 
     lines = output.strip().strip("\n").split("\n")
     #history=""
@@ -86,8 +89,11 @@ def generate(prompt, history,max_new_tokens,temperature=temperature,top_p=top_p,
             print(line)
         if line.startswith("5. "):
             print(line)            
-    return "",[history,(prompt,output)]
-
+    
+    if history:
+        return "",[history,[(prompt,output)]]
+    else:
+        return "",[(prompt,output)]
 
 def clear_fn():
     return None,None