Spaces:

Puree
/

Lab_10

Runtime error

Puree commited on Oct 8, 2022

Commit

105a492

1 Parent(s): 28c81fa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ Original file is located at
 import streamlit as st
 from transformers import pipeline
-classifier = pipeline("image-to-text", model="microsoft/trocr-base-printed")
 def main():
     st.title("Yelp review")

 import streamlit as st
 from transformers import pipeline
+classifier = pipeline("image-classification", model="sentence-transformers/all-MiniLM-L6-v2")
 def main():
     st.title("Yelp review")