Update app.py

app.py

@@ -1,47 +1,33 @@
 import gradio as gr
 import json
-from transformers import pipeline
-from huggingface_hub import InferenceApi
+from huggingface_hub import InferenceClient
 
-# Set up the Inference API to use a GPU in the Hugging Face Space
-inference = InferenceApi(repo_id="gpt-3.5-turbo", task="text-generation", use_gpu=True)
+# Initialize the InferenceClient, specifying the model and enabling GPU
+client = InferenceClient(model="gpt-3.5-turbo", use_gpu=True)
 
 # Function to attempt to fix JSON using a Hugging Face model with GPU
 def ai_fix_json(json_data):
-    # Prepare a prompt for the AI model to fix the JSON
     prompt = f"Fix the following JSON data and make it valid:\n\n{json_data}\n\nFixed JSON:"
-
-    # Generate a response from the model using GPU
-    response = inference(inputs=prompt, max_length=1024, num_return_sequences=1)
-
-    # Extract the fixed JSON from the response (you may need to fine-tune this depending on the model's output format)
+    response = client.text_generation(prompt, max_new_tokens=1024)
+    
     fixed_json = response[0]['generated_text'].split("Fixed JSON:")[-1].strip()
     
-    # Try to load the fixed JSON to ensure it's valid
     try:
         parsed_data = json.loads(fixed_json)
         pretty_json = json.dumps(parsed_data, indent=4)
         return pretty_json, "JSON fixed using AI with GPU."
-    
     except json.JSONDecodeError as e:
         return None, f"Failed to fix JSON: {str(e)}"
 
-# Function to handle file uploads and processing
 def process_file(uploaded_file):
-    # Read the content of the uploaded file directly
     json_data = uploaded_file  # This is already the content of the file as a string
-    
-    # Attempt to fix the JSON using AI
     cleaned_json, message = ai_fix_json(json_data)
     
     if cleaned_json:
-        # Return the fixed JSON for display and as a downloadable file
         return cleaned_json, message, cleaned_json
-    
     else:
         return None, message, None
 
-# Gradio interface
 iface = gr.Interface(
     fn=process_file,
     inputs=gr.File(label="Upload your JSON file"),