Update app.py
Browse files
app.py
CHANGED
|
@@ -9,13 +9,29 @@ from langchain.prompts import PromptTemplate
|
|
| 9 |
from langchain.llms import HuggingFaceHub
|
| 10 |
from langchain.chains import LLMChain
|
| 11 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
|
| 13 |
# https://colab.research.google.com/drive/1hrS6_g14EcOD4ezwSGlGX2zxJegX5uNX#scrollTo=NUwUR9U7qkld
|
| 14 |
llm_hf = HuggingFaceHub(
|
| 15 |
-
repo_id="
|
| 16 |
model_kwargs={"temperature":0.9 }
|
| 17 |
)
|
| 18 |
|
|
|
|
|
|
|
| 19 |
text = "Why did the chicken cross the road?"
|
| 20 |
|
| 21 |
output_question_1 = llm_hf(text)
|
|
|
|
| 9 |
from langchain.llms import HuggingFaceHub
|
| 10 |
from langchain.chains import LLMChain
|
| 11 |
|
| 12 |
+
####
|
| 13 |
+
## Sentinent models
|
| 14 |
+
# https://huggingface.co/spaces/CK42/sentiment-model-comparison
|
| 15 |
+
# 1, 4 seem best for german
|
| 16 |
+
####
|
| 17 |
+
model_id_1 = "nlptown/bert-base-multilingual-uncased-sentiment"
|
| 18 |
+
model_id_2 = "microsoft/deberta-xlarge-mnli"
|
| 19 |
+
model_id_3 = "distilbert-base-uncased-finetuned-sst-2-english"
|
| 20 |
+
model_id_4 = "lordtt13/emo-mobilebert"
|
| 21 |
+
model_id_5 = "juliensimon/reviews-sentiment-analysis"
|
| 22 |
+
model_id_6 = "sbcBI/sentiment_analysis_model"
|
| 23 |
+
|
| 24 |
+
chat_model_facebook-blenderbot-400M-distill = "facebook/blenderbot-400M-distill"
|
| 25 |
+
chat_model_HenryJJ-vincua-13b = "HenryJJ/vincua-13b"
|
| 26 |
|
| 27 |
# https://colab.research.google.com/drive/1hrS6_g14EcOD4ezwSGlGX2zxJegX5uNX#scrollTo=NUwUR9U7qkld
|
| 28 |
llm_hf = HuggingFaceHub(
|
| 29 |
+
repo_id="chat_model_facebook-blenderbot-400M-distill",
|
| 30 |
model_kwargs={"temperature":0.9 }
|
| 31 |
)
|
| 32 |
|
| 33 |
+
|
| 34 |
+
|
| 35 |
text = "Why did the chicken cross the road?"
|
| 36 |
|
| 37 |
output_question_1 = llm_hf(text)
|