Frenchizer commited on
Commit
4b2b04a
·
verified ·
1 Parent(s): f285d15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -9
app.py CHANGED
@@ -16,15 +16,18 @@ context_tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-mnli")
16
  translation_tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-fr")
17
 
18
  labels = [
19
- "general", "pharma", "legal", "technical", "UI", "user interface", "medicine",
20
- "information technology", "IT", "marketing", "e-commerce", "programming",
21
- "website", "html", "keywords", "food commerce", "personal development",
22
- "literature", "poetry", "physics", "chemistry", "biology", "theater",
23
- "finance", "sports", "education", "politics", "economics", "art", "history",
24
- "music", "gaming", "aerospace", "engineering", "robotics", "travel", "tourism",
25
- "healthcare", "psychology", "environment", "fashion", "design", "real estate",
26
- "retail", "news", "entertainment", "social media", "automotive", "machine learning",
27
- "cryptocurrency", "blockchain", "philosophy", "anthropology", "archaeology", "data science"
 
 
 
28
  ]
29
 
30
  def softmax_with_temperature(logits, temperature=1.0):
 
16
  translation_tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-fr")
17
 
18
  labels = [
19
+ "aerospace", "anatomy", "anthropology", "art",
20
+ "automotive", "blockchain", "biology", "chemistry",
21
+ "cryptocurrency", "data science", "design", "e-commerce",
22
+ "education", "engineering", "entertainment", "environment",
23
+ "fashion", "finance", "food commerce", "general",
24
+ "gaming", "healthcare", "history", "html",
25
+ "information technology", "IT", "keywords", "legal",
26
+ "literature", "machine learning", "marketing", "medicine",
27
+ "music", "personal development", "philosophy", "physics",
28
+ "politics", "poetry", "programming", "real estate", "retail",
29
+ "robotics", "slang", "social media", "sports", "sustained",
30
+ "technical", "theater", "tourism", "travel"
31
  ]
32
 
33
  def softmax_with_temperature(logits, temperature=1.0):