update file 037

app.py

@@ -13,8 +13,8 @@ from io import StringIO, BytesIO
 app = Flask(__name__)
 
 # Load model (CPU)
-#generator = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B", device=-1)
-generator = pipeline("text-generation", model="bigscience/bloomz-560m", device=-1)
+generator = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B", device=-1)
+# generator = pipeline("text-generation", model="bigscience/bloomz-560m", device=-1)
 
 # Directories
 os.makedirs("logs", exist_ok=True)
@@ -84,7 +84,7 @@ def generate_ticket():
         issue=issue
     )
 
-    raw = generator(prompt,max_new_tokens=200,do_sample=True, top_p=0.9,temperature=0.8,return_full_text=False)[0]["generated_text"]
+    raw = generator(prompt,max_new_tokens=250,do_sample=True, top_p=0.9,return_full_text=False)[0]["generated_text"]
 
      # Strip the prompt itself, leaving only the new content
     if raw.startswith(prompt):