IBounhas commited on
Commit
abef8be
·
1 Parent(s): 531183f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
3
  import torch
 
4
 
5
  # Define a function that takes a text input and returns the result
6
  def analyze_text(input):
 
1
  import gradio as gr
2
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
3
  import torch
4
+ from sentence_transformers import SentenceTransformer, models
5
 
6
  # Define a function that takes a text input and returns the result
7
  def analyze_text(input):