binqiangliu commited on
Commit
deea6e3
β€’
1 Parent(s): f37ab77

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -3
app.py CHANGED
@@ -1,9 +1,17 @@
1
  from transformers import pipeline, Conversation
2
  import gradio as gr
3
 
4
- #chatbot = pipeline(model="facebook/blenderbot-400M-distill")
5
- chatbot = pipeline(model="HuggingFaceH4/starchat-beta")
6
- #chatbot = pipeline(model="facebook/blenderbot-400M-distill") #Working!
 
 
 
 
 
 
 
 
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 = []