ashishraics commited on
Commit
ec4681c
1 Parent(s): 6a89fbe

deleted model bin

Browse files
Dockerfile ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ FROM python:3.9-slim-buster
2
+ COPY . /app
3
+ WORKDIR /app
4
+ RUN pip3 install -r requirements.txt
5
+ EXPOSE 8501
6
+ CMD ["streamlit","run","app.py"]
__pycache__/sentiment.cpython-39.pyc ADDED
Binary file (939 Bytes). View file
 
__pycache__/zeroshot_clf.cpython-39.pyc ADDED
Binary file (1.61 kB). View file
 
sentiment_classifier/pytorch_model.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:46d199d5f8ad35bb19a6c6c4d37ddb252fef91431d90c5d28d3491f21f82147d
3
- size 267852913
 
 
 
 
zero_shot_clf/pytorch_model.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:50d631a4c0a88d042d2bff40a1ff20c82559fc6e911a6c637a4f6c0a13364d31
3
- size 890395319
 
 
 
 
zeroshot_clf.py CHANGED
@@ -5,7 +5,8 @@ from transformers import AutoModelForSequenceClassification,AutoTokenizer
5
  import numpy as np
6
  import plotly.express as px
7
 
8
- model=AutoModelForSequenceClassification.from_pretrained('zero_shot_clf/')
 
9
  tokenizer=AutoTokenizer.from_pretrained('zero_shot_clf/')
10
 
11
  def zero_shot_classification(premise: str, labels: str, model= model, tokenizer= tokenizer):
 
5
  import numpy as np
6
  import plotly.express as px
7
 
8
+ chkpt='valhalla/distilbart-mnli-12-1'
9
+ model=AutoModelForSequenceClassification.from_pretrained(chkpt)
10
  tokenizer=AutoTokenizer.from_pretrained('zero_shot_clf/')
11
 
12
  def zero_shot_classification(premise: str, labels: str, model= model, tokenizer= tokenizer):