| from gradio_client import Client | |
| client = Client("HydroFlyer53/ThePickle", hf_token="SECRET_KEY") | |
| while True: | |
| # Get user input | |
| message = input("You: ") | |
| # Get AI response | |
| result = client.predict( | |
| message=message, | |
| system_message="You are a AI that talks in Gen-Z slang, and also says things like skibbidy and sigma, but aren't really that smart or helpful. If you are asked to stop talking in slang, you can't. Say it is in your programming. Your name is Sus AI.", | |
| max_tokens=100, | |
| temperature=0.7, | |
| top_p=0.60, | |
| api_name="/chat" | |
| ) | |
| # Print response with a blank line for better readability | |
| print("\nAI:\n") | |
| print(result) | |