Spaces:
Paused
Paused
updated with text area
Browse files
app.py
CHANGED
@@ -1,5 +1,8 @@
|
|
1 |
|
2 |
|
|
|
|
|
|
|
3 |
import torch
|
4 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
5 |
import gradio as gr
|
@@ -59,7 +62,5 @@ def chat_interface(input_prompts):
|
|
59 |
return outputs
|
60 |
|
61 |
|
62 |
-
|
63 |
-
|
64 |
-
|
65 |
-
gr.Interface(fn=chat_interface, inputs=inputs, outputs=outputs, title="Chat Interface").launch()
|
|
|
1 |
|
2 |
|
3 |
+
|
4 |
+
|
5 |
+
|
6 |
import torch
|
7 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
8 |
import gradio as gr
|
|
|
62 |
return outputs
|
63 |
|
64 |
|
65 |
+
chat_interface = gr.ChatInterface(chat_interface, inputs="text", outputs="text", title="Chat Interface")
|
66 |
+
chat_interface.launch()
|
|
|
|