Update app.py

app.py

@@ -4,13 +4,11 @@ from transformers import pipeline
 
 app = FastAPI()
 
-# Define request model for math operations
 class CalculationRequest(BaseModel):
     a: float
     b: float
     operation: str
 
-# Load a smaller Hugging Face model (example: distilgpt2)
 model = pipeline('text-generation', model='distilgpt2')
 
 @app.post("/calculate")
@@ -32,13 +30,11 @@ def calculate(request: CalculationRequest):
 
     return {"result": result}
 
-# Example endpoint using Hugging Face model
 @app.post("/generate")
 def generate_text(prompt: str):
     generated = model(prompt, max_length=50, clean_up_tokenization_spaces=True)
     return {"generated_text": generated[0]['generated_text']}
 
-# New endpoint for testing math operations
 @app.post("/test_math")
 def test_math(request: CalculationRequest):
     a = request.a
@@ -60,4 +56,4 @@ def test_math(request: CalculationRequest):
 
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    uvicorn.run(app, host="0.0.0.0", port=7860)