acecalisto3 commited on
Commit
5ba31a8
·
verified ·
1 Parent(s): 5001217

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -8,7 +8,8 @@ import logging
8
  import nest_asyncio
9
  from typing import Dict, Any, List, Tuple
10
  import transformers
11
- from transformers import pipeline
 
12
  # Remove OpenAI imports
13
  # from langchain_community.llms import OpenAI
14
  # from langchain.chains import LLMChain
@@ -476,4 +477,4 @@ class ChatInterface:
476
 
477
  if __name__ == "__main__":
478
  chat_interface = ChatInterface()
479
- chat_interface.render_interface()
 
8
  import nest_asyncio
9
  from typing import Dict, Any, List, Tuple
10
  import transformers
11
+ from transformers import pipeline, GPT2Tokenizer # Add GPT2Tokenizer import
12
+
13
  # Remove OpenAI imports
14
  # from langchain_community.llms import OpenAI
15
  # from langchain.chains import LLMChain
 
477
 
478
  if __name__ == "__main__":
479
  chat_interface = ChatInterface()
480
+ chat_interface.render_interface()