Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -69,9 +69,11 @@ def process_text(text):
|
|
| 69 |
|
| 70 |
return text
|
| 71 |
|
|
|
|
| 72 |
# Gradio Function
|
| 73 |
@spaces.GPU
|
| 74 |
def generate(
|
|
|
|
| 75 |
message: str,
|
| 76 |
chat_history: list[tuple[str, str]],
|
| 77 |
max_new_tokens: int = DEFAULT_MAX_NEW_TOKENS,
|
|
@@ -129,7 +131,7 @@ def generate(
|
|
| 129 |
chat_interface = gr.ChatInterface(
|
| 130 |
fn=generate,
|
| 131 |
stop_btn=None,
|
| 132 |
-
inputs=[
|
| 133 |
examples=[
|
| 134 |
["Can you explain briefly to me what is the Python programming language?"],
|
| 135 |
["Could you please provide an explanation about the concept of recursion?"],
|
|
|
|
| 69 |
|
| 70 |
return text
|
| 71 |
|
| 72 |
+
|
| 73 |
# Gradio Function
|
| 74 |
@spaces.GPU
|
| 75 |
def generate(
|
| 76 |
+
model_choice: str,
|
| 77 |
message: str,
|
| 78 |
chat_history: list[tuple[str, str]],
|
| 79 |
max_new_tokens: int = DEFAULT_MAX_NEW_TOKENS,
|
|
|
|
| 131 |
chat_interface = gr.ChatInterface(
|
| 132 |
fn=generate,
|
| 133 |
stop_btn=None,
|
| 134 |
+
inputs=[model_choice = gr.Dropdown(["Storytell", "HF Meta Llama 7b Chat"], label="Choose Model")]
|
| 135 |
examples=[
|
| 136 |
["Can you explain briefly to me what is the Python programming language?"],
|
| 137 |
["Could you please provide an explanation about the concept of recursion?"],
|