Update main.py

main.py

@@ -52,8 +52,8 @@ async def stream(item: dict):
     model = llms[item.llm]
     prefix=model['prefix']
     suffix=model['suffix']
-    nctx =  item['nctx'] if 'nctx' is in item.keys() else model['nctx']
-    max_tokens = item['max_tokens'] if 'max_tokens' is in item.keys() else 512
+    nctx =  item['nctx'] if 'nctx' in item.keys() else model['nctx']
+    max_tokens = item['max_tokens'] if 'max_tokens' in item.keys() else 512
     user="""
     {prompt}"""