evanperez commited on
Commit
c07a7cd
1 Parent(s): 19d9ad5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -10,6 +10,8 @@ from langchain.prompts import PromptTemplate
10
  import os
11
  import json
12
  from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig, TextStreamer, ConversationalPipeline
 
 
13
 
14
  ####CREDIT#####
15
  #Credit to author (Sri Laxmi) of original code reference: SriLaxmi1993
 
10
  import os
11
  import json
12
  from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig, TextStreamer, ConversationalPipeline
13
+ from datasets import ConversationalPipeline
14
+
15
 
16
  ####CREDIT#####
17
  #Credit to author (Sri Laxmi) of original code reference: SriLaxmi1993