nsvanirudh commited on
Commit
993bf97
·
verified ·
1 Parent(s): 23ffad8

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -0
app.py ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ import streamlit as st
3
+ from transformers import pipeline
4
+
5
+ # Use a pipeline as a high-level helper
6
+ from transformers import pipeline
7
+
8
+ pipe = pipeline("feature-extraction", model="minchul/cvlface_adaface_ir101_webface12m", trust_remote_code=True)
9
+
10
+ vis_feat = pipe("https://huggingface.co/datasets/nsvanirudh/test/resolve/main/VIS-frame-1653-used.png")
11
+ print(vis_feat)
12
+ st.write(vis_feat)
13
+ #feature_extractor(text,return_tensors = "pt")[0].numpy().mean(axis=0)
14
+
15
+ #st.write(x, 'squared is', x * x)