binqiangliu commited on
Commit
c120853
·
1 Parent(s): 3f19f00

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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,