dancrvlh commited on
Commit
f31bb11
·
1 Parent(s): 72829d7
Files changed (3) hide show
  1. README.md +3 -3
  2. app.py +0 -2
  3. requirements.txt +2 -2
README.md CHANGED
@@ -1,8 +1,8 @@
1
  ---
2
  title: Languages
3
- emoji: 👀
4
- colorFrom: purple
5
- colorTo: blue
6
  sdk: gradio
7
  sdk_version: 3.50.2
8
  app_file: app.py
 
1
  ---
2
  title: Languages
3
+ emoji: 💬
4
+ colorFrom: white
5
+ colorTo: purple
6
  sdk: gradio
7
  sdk_version: 3.50.2
8
  app_file: app.py
app.py CHANGED
@@ -17,10 +17,8 @@ mapping = {
17
 
18
  def predict(text):
19
  inputs = tokenizer(text, return_tensors="pt")
20
-
21
  outputs = model(**inputs)
22
  predictions = outputs.logits
23
-
24
  return mapping[round(predictions.item())]
25
 
26
  iface = gr.Interface(
 
17
 
18
  def predict(text):
19
  inputs = tokenizer(text, return_tensors="pt")
 
20
  outputs = model(**inputs)
21
  predictions = outputs.logits
 
22
  return mapping[round(predictions.item())]
23
 
24
  iface = gr.Interface(
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- pip
2
  gradio
3
- transforms
4
  torch
 
1
+ pip
2
  gradio
3
+ transformers
4
  torch