Spaces:
Sleeping
Sleeping
Commit
·
c120853
1
Parent(s):
3f19f00
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,7 +2,7 @@ import flask
|
|
| 2 |
from flask import Flask, request, jsonify
|
| 3 |
from langchain import PromptTemplate, LLMChain
|
| 4 |
#from langchain.memory import StreamlitChatMessageHistory
|
| 5 |
-
import numpy as np
|
| 6 |
from langchain.chains import LLMChain
|
| 7 |
from langchain.prompts import PromptTemplate
|
| 8 |
from langchain.memory import ConversationBufferMemory
|
|
@@ -11,7 +11,7 @@ from langchain import HuggingFaceHub
|
|
| 11 |
import os
|
| 12 |
from dotenv import load_dotenv
|
| 13 |
load_dotenv()
|
| 14 |
-
from pathlib import Path
|
| 15 |
from huggingface_hub import InferenceClient
|
| 16 |
from langchain import HuggingFaceHub
|
| 17 |
import requests
|
|
@@ -19,9 +19,10 @@ import requests
|
|
| 19 |
# 初始化Chatbot
|
| 20 |
HUGGINGFACEHUB_API_TOKEN = os.getenv('HUGGINGFACEHUB_API_TOKEN')
|
| 21 |
repo_id = os.environ.get('repo_id')
|
| 22 |
-
port = os.getenv('port')
|
| 23 |
|
| 24 |
llm = HuggingFaceHub(repo_id=repo_id,
|
|
|
|
| 25 |
model_kwargs={"min_length":1024,
|
| 26 |
"max_new_tokens":5632, "do_sample":True,
|
| 27 |
"temperature":0.1,
|
|
|
|
| 2 |
from flask import Flask, request, jsonify
|
| 3 |
from langchain import PromptTemplate, LLMChain
|
| 4 |
#from langchain.memory import StreamlitChatMessageHistory
|
| 5 |
+
#import numpy as np
|
| 6 |
from langchain.chains import LLMChain
|
| 7 |
from langchain.prompts import PromptTemplate
|
| 8 |
from langchain.memory import ConversationBufferMemory
|
|
|
|
| 11 |
import os
|
| 12 |
from dotenv import load_dotenv
|
| 13 |
load_dotenv()
|
| 14 |
+
#from pathlib import Path
|
| 15 |
from huggingface_hub import InferenceClient
|
| 16 |
from langchain import HuggingFaceHub
|
| 17 |
import requests
|
|
|
|
| 19 |
# 初始化Chatbot
|
| 20 |
HUGGINGFACEHUB_API_TOKEN = os.getenv('HUGGINGFACEHUB_API_TOKEN')
|
| 21 |
repo_id = os.environ.get('repo_id')
|
| 22 |
+
#port = os.getenv('port')
|
| 23 |
|
| 24 |
llm = HuggingFaceHub(repo_id=repo_id,
|
| 25 |
+
huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
|
| 26 |
model_kwargs={"min_length":1024,
|
| 27 |
"max_new_tokens":5632, "do_sample":True,
|
| 28 |
"temperature":0.1,
|