Update app.py

app.py

@@ -10,8 +10,8 @@ class CalculationRequest(BaseModel):
     b: float
     operation: str
 
-# Load Hugging Face model (example: a simple text generation model)
-model = pipeline('text-generation', model='gpt2')
+# Load a smaller Hugging Face model (example: distilgpt2)
+model = pipeline('text-generation', model='distilgpt2')
 
 @app.post("/calculate")
 def calculate(request: CalculationRequest):