Spaces:
Runtime error
Runtime error
Commit
·
7d5fe80
1
Parent(s):
960495a
intro text
Browse files
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 |
|