token

app.py

@@ -47,8 +47,6 @@ def generate_skills(job_description):
 
     ft_model.eval()
     with torch.no_grad():
-        # Move model to GPU if not already there
-        ft_model.to("cuda")
         output_tokens = ft_model.generate(**model_input, max_new_tokens=200)[0]
         generated_text = tokenizer.decode(output_tokens, skip_special_tokens=True)
     
@@ -68,4 +66,4 @@ inputs = gr.Textbox(lines=10, label="Job description:", placeholder="Enter or pa
 outputs = gr.Textbox(label="Required skills:", placeholder="The required skills will be displayed here...")
 
 gr.Interface(fn=generate_skills, inputs=inputs, outputs=outputs, title="Job Skills Analysis",
-             description="Paste the job description in the text box below and the model will show the required skills for candidates.").launch(share=True)  # Set share=True to create a public link
+             description="Paste the job description in the text box below and the model will show the required skills for candidates.").launch()  # Remove share=True