Update app.py
Browse files
app.py
CHANGED
@@ -144,12 +144,12 @@ def bot(history,message):
|
|
144 |
yield history
|
145 |
|
146 |
|
147 |
-
@spaces.GPU(
|
148 |
def bot_response(message: str,
|
149 |
history: list,
|
150 |
temperature: float,
|
151 |
max_new_tokens: int
|
152 |
-
) -> str:
|
153 |
"""
|
154 |
Generate a streaming response using the llama3-8b model.
|
155 |
Args:
|
|
|
144 |
yield history
|
145 |
|
146 |
|
147 |
+
@spaces.GPU()
|
148 |
def bot_response(message: str,
|
149 |
history: list,
|
150 |
temperature: float,
|
151 |
max_new_tokens: int
|
152 |
+
) -> str:
|
153 |
"""
|
154 |
Generate a streaming response using the llama3-8b model.
|
155 |
Args:
|