Spaces:

vjkarthikeyan
/

malwareDetection

Runtime error

vjkarthikeyan commited on 26 days ago

Commit

2d2057b

verified ·

1 Parent(s): 71c7cc3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
-MODEL_ID = "Zainab984/malimg"  # <- replace with any model above
 clf = pipeline("image-classification", model=MODEL_ID)
 def predict(img):

 import gradio as gr
 from transformers import pipeline
+MODEL_ID = "fgmckee/malimg_preprocess"  # <- replace with any model above
 clf = pipeline("image-classification", model=MODEL_ID)
 def predict(img):