Update app.py

app.py

@@ -2,11 +2,10 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
 
-# Load a language model (e.g., Llama) for generating alternative XPath suggestions
-tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-2-7b")  # Replace with the actual model
-model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-2-7b")
+# Load a publicly accessible model
+tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-2.7B")
+model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-2.7B")
 
-# Helper function to generate alternative XPaths
 def get_alternative_xpaths(html_structure, original_xpath):
     prompt = f"""
     Given the following HTML structure and an original XPath, generate alternative XPaths that could help locate the same element if it has changed position.
@@ -18,18 +17,15 @@ def get_alternative_xpaths(html_structure, original_xpath):
 
     Suggested alternative XPaths:
     """
-    # Encode and generate with the model
+    # Generate response
     inputs = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(inputs['input_ids'], max_length=200, temperature=0.3)
     response_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-    # Process the response to extract XPath suggestions
     alternative_xpaths = response_text.strip().split('\n')
     return alternative_xpaths
 
-# Gradio interface function for self-healing XPath
 def locate_element_with_self_healing(html_structure, original_xpath):
-    # Simulate the failure to find an element and ask for alternatives
     print("Original XPath failed. Attempting to heal...")
     alternative_xpaths = get_alternative_xpaths(html_structure, original_xpath)
     
@@ -38,7 +34,6 @@ def locate_element_with_self_healing(html_structure, original_xpath):
         "alternative_xpaths": alternative_xpaths
     }
 
-# Define Gradio interface
 interface = gr.Interface(
     fn=locate_element_with_self_healing,
     inputs=[
@@ -52,5 +47,4 @@ interface = gr.Interface(
     description="This tool provides alternative XPaths if the original XPath is not found in the given HTML structure."
 )
 
-# Launch the Gradio interface
 interface.launch()