Kiro0o commited on
Commit
28c59c8
·
verified ·
1 Parent(s): b777049

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ !pip install diffusers transformers torch numpy scipy gradio datasets
2
+ !pip3 install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio===0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
3
+ import torch
4
+ from transformers import AutoModelForSequenceClassification, AutoTokenizer, AutoConfig
5
+ import numpy as np
6
+ from scipy.special import softmax
7
+ import gradio as gr
8
+ torch.cuda.is_available()