KevinXiong2022 commited on
Commit
abf7ffa
·
1 Parent(s): 24a09c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -12
app.py CHANGED
@@ -1,15 +1,9 @@
1
- import gradio as gr
2
- from transformers import pipeline
3
 
4
- pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
 
5
 
6
- def predict(image):
7
- predictions = pipeline(image)
8
- return {p["label"]: p["score"] for p in predictions}
9
 
10
- gr.Interface(
11
- predict,
12
- inputs=gr.inputs.Image(label="Upload hot dog candidate", type="filepath"),
13
- outputs=gr.outputs.Label(num_top_classes=2),
14
- title="Hot Dog? Or Not?",
15
- ).launch()
 
1
+ from transformers import BertConfig, BertModel
 
2
 
3
+ # Building the config
4
+ config = BertConfig()
5
 
6
+ # Building the model from the config
7
+ model = BertModel(config)
 
8
 
9
+ print(config)