app.py

Hugging Face's logo
Hugging Face
Search models, datasets, users...
Models
Datasets
Spaces
Docs
Solutions
Pricing



Spaces:

runaksh
/
Symptom-2-disease_Text_Classification


like
0

Logs
App
Files
Community
Settings
Symptom-2-disease_Text_Classification
/
app.py
runaksh's picture
runaksh
Update app.py
7450a2c
about 2 months ago
raw
history
blame
edit
delete
No virus
1.33 kB
# -*- coding: utf-8 -*-
"""gradio_deploy.ipynb
Automatically generated by Colaboratory.
"""
import os
import gradio
from PIL import Image
from timeit import default_timer as timer
from tensorflow import keras
import torch
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
import numpy as np

loaded_model = AutoModelForSequenceClassification.from_pretrained("runaksh/financial_sentiment_distilBERT")
loaded_tokenizer = AutoTokenizer.from_pretrained("runaksh/financial_sentiment_distilBERT")

# Function for class prediction
def predict(sample, validate=True):
  classifier = pipeline("text-classification", model=loaded_model, tokenizer=loaded_tokenizer)
  pred = classifier(sample)[0]['label']
  return pred

title = "Financial Sentiment Classification"
description = "Enter the news"

# Gradio elements

# Input from user
in_prompt = gradio.components.Textbox(lines=2, label='Enter the News')

# Output response
out_response = gradio.components.Textbox(label='Sentiment')

# Gradio interface to generate UI link
iface = gradio.Interface(fn=predict,
                         inputs = in_prompt,
                         outputs = out_response,
                         title=title,
                         description=description
                         )

iface.launch(debug = True)