rajeshradhakrishnan commited on
Commit
6c59131
·
1 Parent(s): 6e6578d

torch and io fix in gradio

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,4 +1,5 @@
1
  from pathlib import Path
 
2
  import gradio as gr
3
  from huggingface_hub import from_pretrained_fastai
4
 
@@ -18,11 +19,11 @@ def predict(im):
18
 
19
  interface = gr.Interface(
20
  predict,
21
- inputs="newsheadlines",
22
  outputs='label',
23
  theme="huggingface",
24
  title="Malayalam News Classifier",
25
- description="Try to classify news in മലയാളം? Input a few malayalam news headlines and verify whether the model categorized it apporpirately!",
26
  article = "<p style='text-align: center'>Malayalam News Classifier | Demo Model</p>",
27
  live=True)
28
  interface.launch(debug=True)
 
1
  from pathlib import Path
2
+ import torch
3
  import gradio as gr
4
  from huggingface_hub import from_pretrained_fastai
5
 
 
19
 
20
  interface = gr.Interface(
21
  predict,
22
+ inputs="textbox",
23
  outputs='label',
24
  theme="huggingface",
25
  title="Malayalam News Classifier",
26
+ description="Try to classify news in മലയാളം? Input a few malayalam news headlines and verify whether the model categorized it appropriately!",
27
  article = "<p style='text-align: center'>Malayalam News Classifier | Demo Model</p>",
28
  live=True)
29
  interface.launch(debug=True)