Update main.py

main.py

@@ -1,9 +1,9 @@
-from fastapi import FastAPI, HTTPException, Request
 import requests
+from fastapi import FastAPI, HTTPException, Query
 
 app = FastAPI()
 
-API_URL = "https://curated.aleph.cloud/vm/b950fef19b109ef3770c89eb08a03b54016556c171b9a32475c085554b594c94/completion"
+API_URL = "https://huggingface.co/spaces/Gemechis/noname/generate-response"
 
 def generate_text(prompt):
     params = {
@@ -24,12 +24,9 @@ def generate_text(prompt):
 @app.get('/')
 def welcome():
     return "Welcome dude"
-
-
+    
 @app.get('/generate-response')
-async def generate_response(request: Request):
-    data = await request.json()
-    prompt = data.get('prompt')
+async def generate_response(prompt: str = Query(..., description="The prompt for text generation")):
     if not prompt:
         raise HTTPException(status_code=400, detail="Prompt is missing")