Update utils.py

utils.py

@@ -1,4 +1,4 @@
-from huggingface_hub import InferenceClient
+rom huggingface_hub import InferenceClient
 from dotenv import load_dotenv
 import os
 import pandas as pd
@@ -46,15 +46,9 @@ Answer (with explanation):
 
     # Use Hugging Face InferenceClient directly
     client = InferenceClient(token=os.environ.get("HUGGINGFACE_API_KEY"))
-    response = client(model="google/gemma-2b-it", inputs=prompt)
-
-    # Depending on the model, you may need to extract the text
-    # Some models return {"generated_text": "..."} or a list of dicts
-    if isinstance(response, list) and "generated_text" in response[0]:
-        answer = response[0]["generated_text"]
-    elif isinstance(response, dict) and "generated_text" in response:
-        answer = response["generated_text"]
-    else:
-        answer = str(response)
+    response = client.text_generation(model="google/gemma-2b-it", inputs=prompt)
+
+    # Extract the generated text from the response
+    answer = response[0]["generated_text"] if isinstance(response, list) and "generated_text" in response[0] else str(response)
 
     return answer