Update main.py

main.py

@@ -70,9 +70,9 @@ async def root(item: Item):
 
         def event_generator():
             for event in replicate.stream("snowflake/snowflake-arctic-instruct", input={
-                "prompt": "summarize this following news article:" + article,
+                "prompt": f"summarize this news article in {item.max_tokens} lines:" + article,
                 "temperature": 0.2,
-                "max_new_tokens" : item.max_tokens
+                "max_new_tokens" : 1000
             }):
                 # Yield the event as a string
                 yield str(event)