Spaces:
Runtime error
Runtime error
| import requests | |
| import json | |
| # Test the API endpoints | |
| base_url = "http://localhost:7860" | |
| def test_api(): | |
| # Test data | |
| test_texts = [ | |
| "I love this product! It's amazing!", | |
| "This is terrible and disappointing.", | |
| "It's okay, nothing special.", | |
| "Best purchase ever! Highly recommend!" | |
| ] | |
| print("🧪 Testing Sentiment Analysis API") | |
| print("=" * 50) | |
| # Test single prediction | |
| print("\n📊 Testing /predict endpoint:") | |
| for text in test_texts[:2]: | |
| try: | |
| response = requests.post( | |
| f"{base_url}/predict", | |
| json={"text": text} | |
| ) | |
| result = response.json() | |
| print(f"Text: '{text}'") | |
| print(f"Prediction: {result['prediction']} ({result['sentiment']})") | |
| print(f"Confidence: {result['confidence']:.2%}") | |
| print("-" * 30) | |
| except Exception as e: | |
| print(f"Error testing '{text}': {e}") | |
| # Test probability prediction | |
| print("\n📈 Testing /predict_proba endpoint:") | |
| test_text = test_texts[0] | |
| try: | |
| response = requests.post( | |
| f"{base_url}/predict_proba", | |
| json={"text": test_text} | |
| ) | |
| result = response.json() | |
| print(f"Text: '{test_text}'") | |
| print(f"Probabilities: {result['probabilities']}") | |
| print(f"Prediction: {result['prediction']} ({result['sentiment']})") | |
| print("-" * 30) | |
| except Exception as e: | |
| print(f"Error testing probabilities: {e}") | |
| # Test batch prediction | |
| print("\n📦 Testing /batch_predict endpoint:") | |
| try: | |
| response = requests.post( | |
| f"{base_url}/batch_predict", | |
| json=test_texts | |
| ) | |
| results = response.json() | |
| for result in results['results']: | |
| print(f"Text: '{result['text']}'") | |
| print(f"Sentiment: {result['sentiment']} (confidence: {result['confidence']:.2%})") | |
| print("-" * 30) | |
| except Exception as e: | |
| print(f"Error testing batch prediction: {e}") | |
| if __name__ == "__main__": | |
| print("Make sure to start the API server first with: python app.py") | |
| print("Then run this test script.") | |
| # Uncomment the line below to run tests (make sure API is running first) | |
| # test_api() | |