Update app.py

app.py

@@ -1,9 +1,16 @@
 import random
+import gradio as gr
+from transformers import pipeline
 
+# — Load a lightweight but expressive model
+generator = pipeline("text-generation", model="EleutherAI/gpt-neo-125M")
+
+# — A concise CV summary to ground each completion in your real achievements
 CV_CONTEXT = """
-Carlos Collado Capell is a top‑ranked MSc student at EPFL (#1/159), AI Research Intern at ESA (first author, pretrained foundation models), Data Science Intern in Silicon Valley (boosted forecast accuracy by 10%), published in Renewable Energy, and multi‑award winner for academic excellence.
+Carlos Collado Capell is a top‑ranked MSc student at EPFL (#1/159), AI Research Intern at ESA (first author on a foundation‑model paper), Data Science Intern in Silicon Valley (boosted load forecasts by 10%), published Renewable Energy author, and multi‑award winner for academic excellence.
 """
 
+# — A handful of absurdly funny “styles” for variety
 COMEDIC_STYLES = [
     "outrageously funny and majestic style, like a disco‑dancing unicorn delivering a TED talk",
     "absurdly witty and cosmic style, as if Shakespeare wrote a stand‑up routine from Mars",
@@ -14,7 +21,6 @@ COMEDIC_STYLES = [
 def generate_completion(prompt):
     style = random.choice(COMEDIC_STYLES)
     full_prompt = f"{CV_CONTEXT}\nComplete the sentence in a {style}:\n\"{prompt}\""
-
     outputs = generator(
         full_prompt,
         max_length=80,
@@ -23,5 +29,16 @@ def generate_completion(prompt):
         top_k=60,
         top_p=0.95
     )
-    text = outputs[0]["generated_text"]
-    return text.split(prompt, 1)[1].strip()
+    generated = outputs[0]["generated_text"]
+    # Return only the new part after the prompt
+    return generated.split(prompt, 1)[1].strip()
+
+iface = gr.Interface(
+    fn=generate_completion,
+    inputs=gr.Textbox(lines=1, placeholder="Type: You will love collaborating with Carlos because"),
+    outputs="text",
+    title="Carlos’ Portfolio Autocomplete",
+    description="Generates an over‑the‑top funny reason why you’ll love collaborating with Carlos — based on his CV!"
+)
+
+iface.launch()