Spaces:

ccm
/

chat-with-SFF

Sleeping

ccm commited on Oct 31, 2024

Commit

9cdb1a5

verified ·

1 Parent(s): c450b02

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -77,7 +77,7 @@ def reply(message: str, history: list[str]) -> str:
     # Preprocess message
     pipe = transformers.pipeline(
-        "text2text-generation", model="Qwen/Qwen2.5-7B-Instruct", device="cuda"
     )
     message = preprocess(message, PUBLICATIONS_TO_RETRIEVE)

     # Preprocess message
     pipe = transformers.pipeline(
+        "text-generation", model="Qwen/Qwen2.5-7B-Instruct", device="cuda"
     )
     message = preprocess(message, PUBLICATIONS_TO_RETRIEVE)