Update app.py
Browse files
app.py
CHANGED
@@ -6,6 +6,8 @@ import pickle
|
|
6 |
import numpy as np
|
7 |
import torch.nn.functional as F
|
8 |
from accelerate import init_empty_weights, infer_auto_device_map, load_checkpoint_and_dispatch
|
|
|
|
|
9 |
|
10 |
# ---- Constants and Setup ----
|
11 |
model_name = 'gpt2'
|
@@ -86,14 +88,6 @@ def advanced_agi_chat(user_input):
|
|
86 |
return response
|
87 |
|
88 |
# ---- Gradio Interface ----
|
89 |
-
def chat_interface(user_input):
|
90 |
-
response = advanced_agi_chat(user_input)
|
91 |
-
return response
|
92 |
-
|
93 |
-
# ---- Gradio App Setup ----
|
94 |
-
import gradio as gr
|
95 |
-
import datetime
|
96 |
-
|
97 |
with gr.Blocks() as app:
|
98 |
gr.Markdown("# **Gertrude**")
|
99 |
|
@@ -203,10 +197,5 @@ with gr.Blocks() as app:
|
|
203 |
queue=True
|
204 |
)
|
205 |
|
206 |
-
|
207 |
-
|
208 |
-
# Setting the button click event
|
209 |
-
|
210 |
-
|
211 |
# Launch the Gradio app
|
212 |
app.launch()
|
|
|
6 |
import numpy as np
|
7 |
import torch.nn.functional as F
|
8 |
from accelerate import init_empty_weights, infer_auto_device_map, load_checkpoint_and_dispatch
|
9 |
+
import gradio as gr
|
10 |
+
import datetime
|
11 |
|
12 |
# ---- Constants and Setup ----
|
13 |
model_name = 'gpt2'
|
|
|
88 |
return response
|
89 |
|
90 |
# ---- Gradio Interface ----
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
91 |
with gr.Blocks() as app:
|
92 |
gr.Markdown("# **Gertrude**")
|
93 |
|
|
|
197 |
queue=True
|
198 |
)
|
199 |
|
|
|
|
|
|
|
|
|
|
|
200 |
# Launch the Gradio app
|
201 |
app.launch()
|