Update app.py

app.py

@@ -20,14 +20,27 @@ os.environ["HUGGINGFACE_API_KEY"] = hf_token
 
 
 # Define the HuggingFaceInferenceWrapper class correctly
+import json
+
 class HuggingFaceInferenceWrapper:
     def __init__(self, inference_api):
         self.inference_api = inference_api
 
     def generate(self, prompt: str, **kwargs) -> str:
-        # Call inference API - returns string directly
-        response = self.inference_api(inputs=prompt)
-        return response.strip()
+        # Request raw response and handle JSON decoding
+        response = self.inference_api(inputs=prompt, raw_response=True)
+        result_json = response.json()
+        
+        # Handle both plain text and chat-style model formats
+        if isinstance(result_json, dict):
+            if "generated_text" in result_json:
+                return result_json["generated_text"].strip()
+            elif "outputs" in result_json and isinstance(result_json["outputs"], str):
+                return result_json["outputs"].strip()
+        
+        # If unknown format, fallback
+        return str(result_json)
+
 
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):