Update app.py

app.py

@@ -1,14 +1,14 @@
 # 测试文本模型
 # import part
 import streamlit as st
-from transformers import pipeline
-import torch
-pipe = pipeline("text-classification", model="distilbert-base-uncased")
-print(pipe("This is a test sentence."))
+# from transformers import pipeline
+# import torch
+# pipe = pipeline("text-classification", model="distilbert-base-uncased")
+# print(pipe("This is a test sentence."))
 
-# 测试视觉模型
-from PIL import Image
-import requests
-pipe = pipeline("image-classification", model="google/vit-base-patch16-224")
-url = "http://images.cocodataset.org/val2017/000000039769.jpg"
-print(pipe(Image.open(requests.get(url, stream=True).raw)))
+# # 测试视觉模型
+# from PIL import Image
+# import requests
+# pipe = pipeline("image-classification", model="google/vit-base-patch16-224")
+# url = "http://images.cocodataset.org/val2017/000000039769.jpg"
+# print(pipe(Image.open(requests.get(url, stream=True).raw)))