Commit
·
deea6e3
1
Parent(s):
f37ab77
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,9 +1,17 @@
|
|
| 1 |
from transformers import pipeline, Conversation
|
| 2 |
import gradio as gr
|
| 3 |
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
|
| 8 |
message_list = []
|
| 9 |
response_list = []
|
|
|
|
| 1 |
from transformers import pipeline, Conversation
|
| 2 |
import gradio as gr
|
| 3 |
|
| 4 |
+
chatbot = pipeline(model="facebook/blenderbot-400M-distill") #Working!
|
| 5 |
+
#https://huggingface.co/facebook/blenderbot-400M-distill/tree/main
|
| 6 |
+
#这个模型文件大小:730MB或1.46GB
|
| 7 |
+
|
| 8 |
+
#chatbot = pipeline(model="HuggingFaceH4/starchat-beta")
|
| 9 |
+
#https://huggingface.co/HuggingFaceH4/starchat-beta/tree/main
|
| 10 |
+
#由于这个模型太大了(9.96+9.86+9.86+1.36GB),会导致如下错误:
|
| 11 |
+
#Runtime error
|
| 12 |
+
#Memory limit exceeded (16Gi)
|
| 13 |
+
|
| 14 |
+
#chatbot = pipeline(model="...")
|
| 15 |
|
| 16 |
message_list = []
|
| 17 |
response_list = []
|