Upload app.py

app.py

@@ -1,12 +1,8 @@
 from fastapi import FastAPI
 from pydantic import BaseModel
-from transformers import pipeline
 
 app = FastAPI(title="SAM_MedTesting")
 
-# Load your model once at startup
-generator = pipeline("text-generation", model="gpt2")
-
 class GenerationRequest(BaseModel):
     prompt: str
     max_new_tokens: int = 50
@@ -16,8 +12,8 @@ class GenerationResponse(BaseModel):
 
 @app.post("/generate", response_model=GenerationResponse)
 def generate(req: GenerationRequest):
-    out = generator(req.prompt, max_length=req.max_new_tokens, do_sample=True)
-    return GenerationResponse(generated_text=out[0]["generated_text"])
+    out = f"hello world: {req}"
+    return out
 
 @app.get("/health")
 def health():