Spaces:

yhamidullah
/

custom-classifier-demo

Running

yhamidullah commited on 4 days ago

Commit

ead95c9

verified ·

1 Parent(s): 3505b99

update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import torch
 import gradio as gr
 from transformers import AutoConfig
-from your_model_code import CustomClassifier
 MODEL_ID = "your-username/custom-classifier-demo"
 config = AutoConfig.from_pretrained(MODEL_ID)

 import torch
 import gradio as gr
 from transformers import AutoConfig
+from models import CustomClassifier
 MODEL_ID = "your-username/custom-classifier-demo"
 config = AutoConfig.from_pretrained(MODEL_ID)