Chris4K commited on
Commit
a6201ac
1 Parent(s): 5f4728f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -1
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="HenryJJ/vincua-13b",
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)