Spaces:
Sleeping
Sleeping
update
Browse files- app.py +1 -0
- 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
|