test private model

main.py

@@ -6,17 +6,22 @@ from transformers import pipeline
 
 varibale = os.getenv('test_variable')
 secret = os.getenv('test_secret')
+secret = os.getenv('HF_Token')
+
 
 
 app = FastAPI()
 
-pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
+# pipe_flan = pipeline("text2text-generation", model="google/flan-t5-small")
+pipe_flan = pipeline("text2text-generation", model="GooooGLE/testVulModel")
+
+
 
 @app.get("/infer_t5")
 def t5(input):
     output = pipe_flan(input)
-    # return {"output": output[0]["generated_text"]}
-    return {"output": str(varibale)+str(secret)}
+    return {"output": output[0]["generated_text"]}
+    # return {"output": str(varibale)+str(secret)}
 
 
 app.mount("/", StaticFiles(directory="static", html=True), name="static")