CreitinGameplays
commited on
Commit
•
66a7808
1
Parent(s):
9f43303
Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import gradio as gr
|
2 |
|
3 |
-
def conversation(prompt="",
|
4 |
# Integrate your Bloom 3b model here to generate response based on prompt and max_tokens
|
5 |
# Replace this with the actual call to your Bloom 3b model
|
6 |
gr.load("models/CreitinGameplays/bloom-3b-conversational")
|
|
|
1 |
import gradio as gr
|
2 |
|
3 |
+
def conversation(prompt="", max_new_tokens=128):
|
4 |
# Integrate your Bloom 3b model here to generate response based on prompt and max_tokens
|
5 |
# Replace this with the actual call to your Bloom 3b model
|
6 |
gr.load("models/CreitinGameplays/bloom-3b-conversational")
|