Spaces:

tensorkelechi
/

signforge

Sleeping

tensorkelechi commited on Oct 23, 2024

Commit

2873e20

verified ·

1 Parent(s): 6010aaa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -89,7 +89,7 @@ def load_pipeline():
     # model = pipeline('image-classification', model=model_id, device='cpu')
     # model.eval()
     print(model)
-    return model#.to('cpu')
 # Streamlit app UI template
 st.title("Signature Forgery Detection")

     # model = pipeline('image-classification', model=model_id, device='cpu')
     # model.eval()
     print(model)
+    return model[0] #.to('cpu')
 # Streamlit app UI template
 st.title("Signature Forgery Detection")