RelativelyUnique commited on
Commit
7d5fe80
1 Parent(s): 960495a

intro text

Browse files
Files changed (1) hide show
  1. app.py +16 -0
app.py CHANGED
@@ -4,6 +4,22 @@ import transformers
4
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
5
  from transformers import pipeline
6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  model = AutoModelForSequenceClassification.from_pretrained('mrcoombes/distilbert-wikipedia-pokemon')
8
  tokenizer = AutoTokenizer.from_pretrained('distilbert-base-uncased')
9
 
 
4
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
5
  from transformers import pipeline
6
 
7
+ title = "Description to Pokemon-Type"
8
+ description = """
9
+ <center>
10
+ Given a written description of a character, real or imagined, living or dead - this bot imagines them as a pokemon
11
+ and then describes which pokemon type they would be. For example...
12
+
13
+ "Born and raised in the Austrian Empire, Nikolai Tesla studied engineering and physics in the 1870s without receiving a degree, gaining practical experience in the early 1880s working in telephony and at Continental Edison in the new electric power industry."
14
+ would be an electric-type pokemon.
15
+
16
+ </center>
17
+ """
18
+
19
+ article = "Inspired by [this article](https://medium.com/analytics-vidhya/predicting-pok%C3%A9mon-type-with-the-pok%C3%A9dex-7038754dc422)."
20
+
21
+
22
+
23
  model = AutoModelForSequenceClassification.from_pretrained('mrcoombes/distilbert-wikipedia-pokemon')
24
  tokenizer = AutoTokenizer.from_pretrained('distilbert-base-uncased')
25