Update app.py

app.py

@@ -9,10 +9,15 @@ def main():
     if button:
         st.write(f"Hello, {name}! Nice to meet you!")
 
-        #Load model
+        #Load default class model
         pipe = pipeline("text-classification")        
         results = pipe(name)
         st.write(results)
 
+        #Load custom model from user/repoName
+        pipe2 = pipeline(model="FacebookAI/roberta-large-mnli")
+        results2 = pipe2(name)
+        st.write(results2)
+
 if __name__ == "__main__":
     main()