Update main.py

main.py

@@ -62,18 +62,18 @@ def gpt_with_google_search(prompt):
 
 class Linlada(BaseModel):
     prompt: str
-    web_access: str = False
-    sampler: str = None
-    seed: int = None
-    neg: str = None
+    web_access: str 
+    model: str = 'gpt-3.5-turbo'
+
 
 
 @app.post('/linlada')
 def linlada(request: Linlada):
-    llm = G4F(model=model)
     prompt = request.prompt
+    model = request.model
     web_access = request.web_access
-    if web_access == True:
+    llm = G4F(model=model)
+    if web_access == "true":
        chat = gpt_with_google_search(prompt)
     else:
         chat = llm(prompt)