AiLab / model.py
AGofficial's picture
Upload 3 files
533e375 verified
'''
You can change the AI model by modifying the function 'AskAI()' below.
'''
import requests
import urllib.parse
class AIModel:
def AskAI(self, prompt: str) -> str:
base_url = "https://text.pollinations.ai/"
encoded_prompt = urllib.parse.quote(prompt)
full_url = base_url + encoded_prompt
response = requests.get(full_url)
if response.status_code == 200:
return response.text
else:
return f"<error> | Error: {response.status_code} - {response.text}"
if __name__ == "__main__":
model = AIModel()
print(model.AskAI("Tell me the capital of France, please."))