Update app.py

app.py

@@ -1,26 +1,3 @@
-# from fastapi import FastAPI
-# from fastapi.responses import JSONResponse
-# from fastapi import Request
-# from huggingface_hub import InferenceClient
-
-# app = FastAPI()
-
-# @app.post("/")
-# async def greet_json(request: Request):
-#     input_data = await request.json()
-#     # number = input_data.get("number")
- 
-#     # tripled_number = number * 2
-#     # return {"message": f"Your input number is: {number}, your doubled number is: {tripled_number}"}
-#     user_input = input_data.get("user_input")
-    
-#     client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
-#     # Get the response from the model
-#     response = client(user_input)
-    
-#     # assistant_response = client.text_generation(user_input)
-#     assistant_response = "I am assistant."
-#     return {"assistant_message": f"Your input message is: {user_input}, assistant_response is: {response}"}
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from huggingface_hub import InferenceClient
@@ -32,7 +9,6 @@ app = FastAPI()
 
 # Get the token from the environment variable
 hf_token = os.environ.get("HF_TOKEN")
-login(hf_token)
 
 # client = InferenceClient("HuggingFaceH4/zephyr-7b-beta", token=hf_token)
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta", token=hf_token)