Spaces:

Kr08
/

Llama

Sleeping

Kr08 commited on Sep 3

Commit

498e38f

•

1 Parent(s): 9d868c2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,8 @@ pipeline = transformers.pipeline(
 @spaces.GPU
 def generate_text(input_text):
-    output = pipeline(messages,
                       max_new_tokens=256,
                      )
     # input_tokens = model.tokenizer(input_text,

 @spaces.GPU
 def generate_text(input_text):
+    prompt = {"role": "system", "content": "Summarize the following: "+input_text}
+    output = pipeline(prompt,
                       max_new_tokens=256,
                      )
     # input_tokens = model.tokenizer(input_text,