kookoobau commited on
Commit
1d8154e
1 Parent(s): 9af1238
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +2 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
  from langchain.chains import ConversationChain
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
 
3
 
4
  # Define the LangChain chat agent
5
  model_name = "microsoft/DialoGPT-medium"
 
1
  from langchain.chains import ConversationChain
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
+ import gradio as gr
4
 
5
  # Define the LangChain chat agent
6
  model_name = "microsoft/DialoGPT-medium"
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  langchain==0.0.161
2
  gradio==2.3.0
3
- transformers
 
 
1
  langchain==0.0.161
2
  gradio==2.3.0
3
+ transformers
4
+ torch