CasualCodes commited on
Commit
0ede22e
·
1 Parent(s): 4579668

attempted example transformer test

Browse files
Files changed (1) hide show
  1. appTr.py +27 -0
appTr.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from PIL import Image
3
+ #import tensorflow as tf
4
+ #from tensorflow.keras.models import load_model
5
+ #import numpy as np
6
+
7
+ # Use a pipeline as a high-level helper
8
+ from transformers import pipeline
9
+
10
+ pipe = pipeline("image-to-text", model="Flova/omr_transformer")
11
+
12
+ # # Load model directly
13
+ # from transformers import AutoTokenizer, AutoModel
14
+
15
+ # tokenizer = AutoTokenizer.from_pretrained("Flova/omr_transformer")
16
+ # model = AutoModel.from_pretrained("Flova/omr_transformer")
17
+
18
+ # Using Flova/omr_transformer
19
+
20
+ def notation_2_note(input_img):
21
+ prediction = pipe(Image.fromarray(input_img))
22
+ #print(type(prediction))
23
+ output_text = prediction[0]['generated_text']
24
+ return output_text
25
+
26
+ demo = gr.Interface(notation_2_note, gr.Image(), "text")
27
+ demo.launch()