Update main.py

main.py

@@ -46,20 +46,22 @@ async def parse_raw(request: Request):
     return await request.body()
 
 # Generate text using the Inference API
-def generate_text(prompt: str) -> str:
-    data = {
-        "inputs": prompt,
-        "options": {
-            "max_new_tokens": 200,
-            "temperature": 0.7,
-            "top_p": 0.95,
-            "use_cache": False,
-        },
-    }
+def generate_text(input_text: str) -> str:
+    response = openai.Completion.create(
+        engine="text-davinci-003",
+        prompt=input_text,
+        max_tokens=100,
+        n=1,
+        stop=None,
+        temperature=0.5,
+        top_p=1.0,
+        frequency_penalty=0.0,
+        presence_penalty=0.0
+    )
 
-    response = requests.post(endpoint, headers=headers, json=data)
-    print(response.json())  # Add this line to print the response content
-    return response.json()["generated_text"]
+    print(response)  # Print the entire response object
+
+    return response.choices[0].text.strip()
     
 # Route for generating text
 @app.post("/generate_text")