Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ pipe = pipeline("text-generation", model="meta-llama/Llama-3.2-1B-Instruct", tok
|
|
8 |
def predict(system="you are a helpful assistant", query):
|
9 |
messages = []
|
10 |
messages.append({"role": "system","content": system})
|
11 |
-
messages.append({"role": "
|
12 |
return pipe(messages,max_new_tokens=256)[0]["generated_text"][-1]
|
13 |
|
14 |
gradio_app = gr.Interface(
|
|
|
8 |
def predict(system="you are a helpful assistant", query):
|
9 |
messages = []
|
10 |
messages.append({"role": "system","content": system})
|
11 |
+
messages.append({"role": "user","content": query})
|
12 |
return pipe(messages,max_new_tokens=256)[0]["generated_text"][-1]
|
13 |
|
14 |
gradio_app = gr.Interface(
|